NMS算法和Soft-NMS(soft non-maximum suppression algorithm)算法修正
使用函数变换替代原有的边界框选取
原变换为:
Soft-NMS函数变换:
非最大值抑制(non-maximum suppression,NMS)算法应用于检测框选择:
检测框集合B,其对应的分数集合S,在S中找到最大值M,每一个B中其它的检测框与最大值M对应的检测框计算IOU,当IOU>某一阈值,移除这个检测框。
Soft-NMS:
其它每一个B中的其它检测框的分数由Soft-NMS函数变换,而不是直接移除(程序中是将分数置零)。这样不会把相互重叠的目标丢失掉。
在检测相互重叠的目标时候会产生漏检,
优点:
1.无需重新训练,直接更改测试代码即可
2.能够有效检测出相互重叠的物体