基于注意力機(jī)制的目標(biāo)檢測(cè)方法、網(wǎng)絡(luò)、設(shè)備和存儲(chǔ)介質(zhì)

基本信息

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

摘要

摘要 本申請(qǐng)涉及圖像處理技術(shù)領(lǐng)域,公開(kāi)了一種基于注意力機(jī)制的目標(biāo)檢測(cè)方法、網(wǎng)絡(luò)、設(shè)備和存儲(chǔ)介質(zhì)。該方法包括:通過(guò)特征提取網(wǎng)絡(luò)提取待檢測(cè)圖像的特征,得到第一特征圖;通過(guò)區(qū)域生成網(wǎng)絡(luò)處理第一特征圖,得到至少一個(gè)候選區(qū)域;通過(guò)ROI池化網(wǎng)絡(luò)處理第一特征圖和候選區(qū)域,得到第二特征圖;通過(guò)分類(lèi)網(wǎng)絡(luò)和回歸網(wǎng)絡(luò)分別處理第二特征圖,得到至少一個(gè)目標(biāo)的類(lèi)別得分和邊界框信息;特征提取網(wǎng)絡(luò)、分類(lèi)網(wǎng)絡(luò)和回歸網(wǎng)絡(luò)中的至少一個(gè)包括注意力子網(wǎng)絡(luò);注意力子網(wǎng)絡(luò)確定在先層和臨時(shí)層;基于在先層和臨時(shí)層,確定注意力權(quán)重;將在先層和注意力權(quán)重進(jìn)行乘積操作,對(duì)在先層增加注意力。通過(guò)上述技術(shù)方案,提高目標(biāo)檢測(cè)的速度和精確度。