非极大值抑制中的DIOU-nms

目录

什么是非极大值抑制？

DIoU NMS

公式

---

什么是非极大值抑制？

非极大值抑制（Non-maximum supression）简称NMS，其作用是去除冗余的检测框，去冗余手段是剔除与极大值重叠较多的检测框结果。

简单来说，非极大值抑制的作用就是模型检测出了很多框，我应该留下哪些。

 核心思想：是搜索目标局部范围内的边界框置信度最大的这个最优值，去除目标邻域内的冗余边界框。

DIoU NMS

论文：   https://arxiv.org/pdf/1911.08287.pdf.

一个成熟的IoU衡量指标应该要考虑预测框与真实框的重叠面积、中心点距离、长宽比三个方面。但是IoU 只考虑到了预测框与真实框重叠区域，并没有考虑到中心点距离、长宽比。

基于此，DIOU-NMS就不仅仅考虑IOU，还考虑两个框中心点之间的距离。如果两个框之间IOU比较大，但是两个框的中心距离比较大时，可能会认为这是两个物体的框而不会被过滤掉。
具体的GIoU、DIoU、CIoU的细节可以看我的以下博客：GIoU、DIoU、CIoU

公式

DIoU (Distance-IoU )。简单地在IoU loss基础上添加一个惩罚项，该惩罚项用于最小化两个bbox的中心点距离。

DIoU公式：

如下图，绿色框代表真实框，黑色框代表预测框，b 为预测框的中心，为真实框的中心，代表真实框与预测框中心距离的平方,c表示两个框的最小闭包区域（同时包含了预测框和真实框的最小矩形框）的对角线长度。