基于注意力機(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è)的速度和精確度。 |
