《YOLOV4&5原理与源代码解析之四:DIoU-NMS》

一、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

CV&DL
原文地址:https://www.cnblogs.com/winslam/p/14436708.html