Learn Before
Non-Maximum Suppression Algorithm
The non-maximum suppression (NMS) algorithm filters out redundant predicted bounding boxes based on their confidence scores and spatial overlap. For a given predicted bounding box , its confidence is the largest predicted class likelihood, denoted as . First, all non-background bounding boxes on an image are sorted by confidence in descending order to form a list . The algorithm iteratively selects the bounding box with the highest confidence in as a basis and removes any remaining boxes in whose Intersection over Union (IoU) with exceeds a predefined threshold . Those with non-maximum confidence scores are thus suppressed. This process is repeated using the next highest confidence box as the new basis until every box in has been used as a basis, resulting in a final set of predictions where no two boxes have an IoU greater than .
0
1
Tags
D2L
Dive into Deep Learning @ D2L