[發明專利]目標檢測方法、裝置、電子設備及存儲介質

在審

• 申請號: 202110496899.0
• 申請日: 2021-05-07
• 公開（公告）號: CN113327226A
• 公開（公告）日: 2021-08-31
• 發明（設計）人: 李建強;謝海華;劉冠杰;張磊
• 申請（專利權）人: 北京工業大學
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06N3/08;G06N3/04
• 代理公司: 北京路浩知識產權代理有限公司 11002
• 代理人: 趙赫

1.一種目標檢測方法，其特征在于，包括：

獲取待檢測圖像；

使用基于多層交叉注意力特征金字塔網絡MCAFPN的目標檢測模型對所述待檢測圖像進行目標檢測，其中將卷積網絡的不同層級、不同分辨率的特征圖輸入MCAFPN，所述MCAFPN使用多層交叉注意力模塊將所述不同層級、不同分辨率的特征圖在空間維度上層層級聯，然后輸出增強后的多層特征圖。

2.根據權利要求1所述的目標檢測方法，其特征在于，所述將卷積網絡的不同層級、不同分辨率的特征圖輸入MCAFPN包括：

將第一特征圖作為MCAFPN第一層特征圖，將所述MCAFPN第一層特征圖與第二特征圖輸入所述多層交叉注意力模塊，輸出MCAFPN第二層特征圖；

將所述MCAFPN第二層特征圖與第三特征圖輸入所述多層交叉注意力模塊，輸出MCAFPN第三層特征圖；

其中所述第一特征圖、第二特征圖和第三特征圖是所述卷積網絡的不同層級、不同分辨率的特征圖，其層數由深到淺，第二特征圖的分辨率是第一特征圖的n倍，第三特征圖的分辨率是第二特征圖的n倍，n大于等于2。

3.根據權利要求2所述的目標檢測方法，其特征在于，所述MCAFPN使用多層交叉注意力模塊將所述不同層級、不同分辨率的特征圖在空間維度上層層級聯，然后輸出增強后的多層特征圖包括：

將做n倍上采樣后的所述MCAFPN第一層特征圖和所述第二特征圖輸入第一層交叉注意力模塊，得到第一層交叉注意力特征圖；

將所述第二特征圖和所述第一層交叉注意力特征圖輸入第二層交叉注意力模塊，得到第二層交叉注意力特征；

將所述第二特征圖與所述第二層交叉注意力特征圖逐像素相加，得到所述MCAFPN第二層特征圖；

其中所述第一層交叉注意力模塊和所述第二層交叉注意力模塊對輸入進行交叉聯積和交叉加權。

4.根據權利要求1所述的目標檢測方法，其特征在于，所述第一層交叉注意力模塊和第二層交叉注意力模塊對輸入進行交叉聯積和交叉加權包括：

計算淺層特征圖任一空間位置與深層特征圖對應交叉區域的空間相關性；

歸一化交叉空間中的特征相關性，得到交叉注意力權重；

基于所述交叉注意力權重，與所述深層特征圖進行交叉注意力加權，得到最終的交叉注意力特征。

5.一種目標檢測裝置，其特征在于，包括：

獲取模塊，用于獲取待檢測圖像；

目標檢測模塊，用于使用基于多層交叉注意力特征金字塔網絡MCAFPN的目標檢測模型對所述待檢測圖像進行目標檢測，其中將卷積網絡的不同層級、不同分辨率的特征圖輸入MCAFPN，所述MCAFPN使用多層交叉注意力模塊將所述不同層級、不同分辨率的特征圖在空間維度上層層級聯，然后輸出增強后的多層特征圖。

6.一種電子設備，包括存儲器、處理器及存儲在存儲器上并可在處理器上運行的計算機程序，其特征在于，所述處理器執行所述程序時實現根據權利要求1至4任一項所述目標檢測方法的步驟。

7.一種非暫態計算機可讀存儲介質，其上存儲有計算機程序，其特征在于，該計算機程序被處理器執行時實現根據權利要求1至4任一項所述目標檢測方法的步驟。