《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