DIoU-NMS

和原始的NMS不同,DIoU-NMS不僅考慮了IoU的值,還考慮了兩個Box中心點之間的距離,使用了新的公式決定一個Box是否被刪除: 其中,RDIoU是兩個Box中心點之間的距離,用下面的公式 ...

NMS及其變體

nms:1.首先將pred_data中置信度小於x的過濾掉，2.根據conf從大到小重新排序並記為P。3.將P[0]保存到另一個列表R中並計算其與P[1:]的IOU(可選項：是否在不同類間計算IOU) ...