基于注意力機制的目標檢測方法、網絡、設備和存儲介質

基本信息

申請?zhí)?/td> CN202010727998.0 申請日 -
公開（公告）號 CN112016569A 公開（公告）日 2020-12-01
申請公布號 CN112016569A 申請公布日 2020-12-01
分類號 G06K9/46;G06K9/32;G06K9/62 分類 計算；推算；計數;
發(fā)明人 任豪;郭鈺 申請（專利權）人 馭勢科技（浙江）有限公司
代理機構 北京開陽星知識產權代理有限公司 代理人 郭鑫
地址 211100 江蘇省南京市江寧開發(fā)區(qū)蘇源大道19號九龍湖國際企業(yè)總部園B4座二層（江寧開發(fā)區(qū)）
法律狀態(tài) -

摘要

摘要 本申請涉及圖像處理技術領域，公開了一種基于注意力機制的目標檢測方法、網絡、設備和存儲介質。該方法包括：通過特征提取網絡提取待檢測圖像的特征，得到第一特征圖；通過區(qū)域生成網絡處理第一特征圖，得到至少一個候選區(qū)域；通過ROI池化網絡處理第一特征圖和候選區(qū)域，得到第二特征圖；通過分類網絡和回歸網絡分別處理第二特征圖，得到至少一個目標的類別得分和邊界框信息；特征提取網絡、分類網絡和回歸網絡中的至少一個包括注意力子網絡；注意力子網絡確定在先層和臨時層；基于在先層和臨時層，確定注意力權重；將在先層和注意力權重進行乘積操作，對在先層增加注意力。通過上述技術方案，提高目標檢測的速度和精確度。