Deformable DETR DETR 存在收斂速度慢等缺陷。為了解決這些問題,本文可變形 DETR,其注意力模塊僅關注於參考點附近的一小部分采樣點作為注意力模塊中的 key 元素。可變形 DETR 可以在比 DETR 少 9/10 的訓練輪數下,達到更好的性能(尤其是在小目標上)。在 COCO 基准上的大量實驗表明了該方法的有效性。


Deformable DETR

 

圖片

論文發布:2020年10月初  |  代碼開源:2020年12月初

作者團隊:商湯研究院, 中科大, 港中文

論文鏈接:https://arxiv.org/abs/2010.04159

項目代碼鏈接:

https://github.com/fundamentalvision/Deformable-DETR

 

DETR 存在收斂速度慢等缺陷。為了解決這些問題,本文可變形 DETR,其注意力模塊僅關注於參考點附近的一小部分采樣點作為注意力模塊中的 key 元素。可變形 DETR 可以在比 DETR 少 9/10 的訓練輪數下,達到更好的性能(尤其是在小目標上)。在 COCO 基准上的大量實驗表明了該方法的有效性。

 

圖片

 

圖片

 

 

圖片

 

想了解這篇論文,詳情可見:訓練加快10倍!性能更強!商湯等提出可變形DETR目標檢測網絡


免責聲明!

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



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