作者丨小馬
編輯丨極市平臺
本文原創(chuàng)首發(fā)于極市平臺公眾號，轉(zhuǎn)載請獲得授權(quán)并標(biāo)明出處。

【寫在前面】

在本文中，作者提出了一種基于Transformer的目標(biāo)檢測器。在以前基于Transformer的檢測器中，目標(biāo)的查詢是一組可學(xué)習(xí)的embedding。然而，每個可學(xué)習(xí)的embedding都沒有明確的意義 （因為是隨機(jī)初始化的），所以也不能解釋它最終將集中在哪里。此外，由于每個對象查詢將不會關(guān)注特定的區(qū)域，所以訓(xùn)練時優(yōu)化也是比較困難的 。

為了解決這些問題，作者借鑒了CNN目標(biāo)檢測器中的Anchor Point機(jī)制，使得每個查詢都基于特定的Anchor Point的。因此，每個查詢都可以集中在Anchor Point附近的目標(biāo)上。此外，本文的查詢設(shè)計可以在一個位置預(yù)測多個目標(biāo)。為了降低attention的計算成本，作者設(shè)計了一種輕量級的attention變體。基于新的查詢設(shè)計和attention變體，本文提出的Anchor DETR可以獲得比DETR更好的性能和更快的運(yùn)行速度。

1. 論文和代碼地址

Anchor DETR: Query Design for Transformer-Based Detector

論文地址：https://arxiv.org/abs/2109.07107

代碼地址：https://github.com/megvii-model/AnchorDETR（即將開源）

原文地址：多快好省的目標(biāo)檢測器來了！曠視孫劍團(tuán)隊提出 Anchor DETR：基于 Anchor Point 的 DETR

總結(jié)

以上是生活随笔為你收集整理的多快好省的目标检测器来了，旷视孙剑团队提出Anchor DETR：基于Anchor Point的DETR的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。