softnms代码解读

NMS算法缺点

NMS算法

• 非极大值抑制算法NMS 存在的两个问题：

• 当阈值过小时，如图所示，绿色框容易被抑制；当过大时，容易造成误检，即抑制效果不明显。Soft-NMS

为解决NMS算法中存在的上述两个问题，Soft-NMS在进行非极大抑制的时同时考虑得分和 边框之间的重合程度。

对于NMS而言，在每次取出得分最大的框后，计算最大框与其他框的重合程度，若重合程度过大则剔除。而Soft-NMS则以一个权重的形式，将获得的IOU取高斯指数后乘上原得分，之后重新排序再继续循环。两者区别代码如下所示。

NMS中边框剔除代码：

Soft-NMS边框剔除代码

Soft-NMS伪代码

Soft-NMS整体代码

Soft-NMS总结

在NMS算法中，直接将IoU 超过阈值的检测框的得分设置为 0，而 soft NMS 则将其得分进行惩罚衰减，有两种衰减方式。第一种是使用1-IoU与得分的乘积作为衰减后的值，但这种方式在略低于阈值和略高于阈值的部分，经过惩罚衰减函数后，很容易导致得分排序的顺序打乱，合理的惩罚函数应该是具有高IoU的有高的惩罚，低IoU的有低的惩罚，它们中间应该是逐渐过渡的，因此提出高斯惩罚函数。

from：https://baijiahao.baidu.com/s?id=1704419893916597186&wfr=spider&for=pc

本文来自博客园，作者：海_纳百川，转载请注明原文链接：https://www.cnblogs.com/chentiao/p/16420710.html，如有侵权联系删除