一、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