Enhancing Object Detection in Dense Images: Adjustable Non-Maximum Suppression for Single-Class Detection

Abstract

Deep learning-based object detection technology often relies on non-maximum suppression (NMS) algorithms to eliminate redundant detections. However, the conventional NMS algorithm struggles with distinguishing between overlapping and small objects due to its simple constraints. While Soft-NMS offers a slight improvement in object detection performance, it still falls short in addressing this challenge. Our proposed solution, adjustable-NMS, represents a significant advancement. While performing comparably to NMS and Soft-NMS on less dense images where objects are easily countable, adjustable-NMS excels in scenarios with higher object density or smaller objects. In such cases, it outperforms both NMS and Soft-NMS, showcasing notably superior object detection capabilities. On average, the improvement achieved with adjustable-NMS reaches an impressive 33.3%. This demonstrates adjustable-NMS’s efficacy in enhancing object detection accuracy, particularly in challenging environments characterized by dense scenes or diminutive objects.

Keywords

• Adjustable-NMS
• intersection over union
• NMS
• object detection
• soft-NMS
• YOLO