《YOLOV4&5原理與源代碼解析之四:DIoU-NMS》


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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM