NMS 与 Soft-NMS
1)NMS
非最大抑制(Non-maximum suppression, NMS)是物体检测流程中重要的组成部分。
NMS算法首先按照得分从高到低对建议框进行排序,然后分数最高的检测框M被选中,其他框与被选中建议框有明显重叠的框被抑制。该过程被不断递归的应用于其余检测框。
根据算法的设计,如果一个物体处于预设的重叠阈值之内,可能会导致检测不到该待检测物体。即当两个目标框接近时,分数更低的框就会因为与之重叠面积过大而被删掉。(将检测框按得分排序,然后保留得分最高的框,同时删除与该框重叠面积大于一定比例的其它框)
例如,下面的二者的得分分别是0.95和0.80。如果按照NMS进行处理,首先选中得分最高的红色框,然后绿色框就会因为与之重叠面积过大而被删掉。
算法如下:
2)Soft-NMS
NMS是将检测框按得分排序,然后保留得分最高的框,同时删除与该框重叠面积大于一定比例的其它框。
而Soft-NMS不是直接将其去掉,而是利用一个权重函数,每次都会衰减与得分最高的框重叠的相邻检测框的检测分数。越是高度重叠,越是衰减的更严重,然后最后如果还是小于设置的阈值,则进行去除。
算法对比如下: