Non Maximum Suppression: Theory and Implementation in PyTorch

Non Maximum Suppression (NMS) is a technique used in numerous computer vision tasks. It is a class of algorithms to select one entity (e.g., bounding boxes) out of many overlapping entities. We can choose the selection criteria to arrive at the desired results. The criteria are most commonly some form of probability number and some form of overlap measure (e.g. This post will go over how it works and implement it in Python using the PyTorch framework.