一、Soft-NMS
先回顾下传统NMS,如下算法流程图:假如检测马,首先检测到置信度最大的红色框,然后遍历红色框附近的框并计算IOU,IOU大的框都删掉。
因为IOU大,表示高度重叠,所以删除。至于SOFT-NMS,由于NMS过于绝对,例如下图,绿框和红框重叠大多,绿框很可能被删除,这样会漏检;而soft-NMS不会删除绿框,他会给绿色框降分,如图从
0.8下降至0.4。
二、DIOU-NMS
注意:有读者会有疑问,这里为什么不用CIOU_nms,而用DIOU_nms?
答:因为前面讲到的CIOU_loss,是在DIOU_loss的基础上,添加的影响因子,包含groundtruth标注框的信息,在训练时用于回归。
但在测试过程中,并没有groundtruth的信息,不用考虑影响因子,因此直接用DIOU_nms即可。
reference:
https://zhuanlan.zhihu.com/p/151914931
DIOU-NMS,讲的比较有水平:https://zhuanlan.zhihu.com/p/105470021
reference:https://zhuanlan.zhihu.com/p/143747206