Improving Object Detection With One Line of Code https://arxiv.org/abs/1704.04503
Code: https://github.com/bharatsingh430/soft-nms
本文针对目标检测,对非极大值抑制这个环节进行了改善,可以检测出两个相邻的相同物体。 对于目标检测问题,我们目前主流的方法过程如下:
如果两个相似物体是相邻的话,以前的非极大值抑制就会只检测出一个目标。
原来的非极大值抑制我们在一个区域只取一个最大的矩形框,附近的矩形框置信度全部清零。这里我们提出根据两个矩形框的重合面积来降低非最大矩形框的置信度。这里我们称之为 Soft-NMS
这里我们用绿色框的代码取代红色框的代码。