和原始的NMS不同,DIoU-NMS不仅考虑了IoU的值,还考虑了两个Box中心点之间的距离,使用了新的公式决定一个Box是否被删除:
其中,RDIoU是两个Box中心点之间的距离,用下面的公式表示:
其中p(.)是距离,b和bgt表示两个box,c是包含两个box的最小box的对角线长度,如下图所示：
