和原始的NMS不同,DIoU-NMS不仅考虑了IoU的值,还考虑了两个Box中心点之间的距离,使用了新的公式决定一个Box是否被删除: 其中,RDIoU是两个Box中心点之间的距离,用下面的公式 ...

nms:1.首先将pred_data中置信度小于x的过滤掉，2.根据conf从大到小重新排序并记为P。3.将P[0]保存到另一个列表R中并计算其与P[1:]的IOU(可选项：是否在不同类间计算IOU) ...