Object Detection: Non-Maximum Suppression (NMS)

Naoki
8 min readJun 1, 2022

Understanding the Logic and Tricky Part of NMS

Object detection models like YOLOv5 and SSD predict objects’ locations by generating bounding boxes (shown in blue rectangles below). However, object detection models produce more bounding boxes than the final output with different locations, sizes, and confidence levels. They do not just predict one bounding box per object.

--

--