Understanding NMS (Non-Maximum Suppression)

Learn how NMS, a post processing logic, is used to eliminate redundant boxes

Why do we need NMS?

Object detection models often predict multiple bounding boxes for a single object in an image. However, the final output should ideally have only one bounding box per object. To achieve this, a technique called non-maximum suppression (NMS) is used.

In the image below, there are three bounding boxes predicted for one person with different confidence scores. On visual inspection, for our final output, we would prefer the green box because it fits better as compared to the other two boxes and has the highest confidence score.

Create a free account to view this lesson.

By signing up, you agree to Educative's Terms of Service and Privacy Policy