一種基于注意力機(jī)制的目標(biāo)分類方法及系統(tǒng)

基本信息

申請?zhí)?/td> CN202110943944.2 申請日 -
公開(公告)號 CN113762251A 公開(公告)日 2021-12-07
申請公布號 CN113762251A 申請公布日 2021-12-07
分類號 G06K9/32(2006.01)I;G06K9/62(2006.01)I;G06N3/04(2006.01)I 分類 計(jì)算;推算;計(jì)數(shù);
發(fā)明人 柴象飛;郭娜;張路;劉鵬飛;秦浩峰 申請(專利權(quán))人 慧影醫(yī)療科技(北京)股份有限公司
代理機(jī)構(gòu) 北京輕創(chuàng)知識產(chǎn)權(quán)代理有限公司 代理人 陳霆雷
地址 100192 北京市海淀區(qū)西小口路66號中關(guān)村東升科技園·北領(lǐng)地B-2號樓2層A206
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種基于注意力機(jī)制的目標(biāo)分類方法,涉及神經(jīng)網(wǎng)絡(luò)技術(shù)領(lǐng)域。包括:獲取待分類圖像的步驟;將待分類圖像輸入到卷積神經(jīng)網(wǎng)絡(luò),提取待分類圖像的特征,通過注意力矩陣塊對得到的特征圖進(jìn)行處理,根據(jù)處理后的特征圖得到至少一個(gè)目標(biāo)分類結(jié)果的步驟。本發(fā)明適用于圖像中目標(biāo)物的分類,通過加入注意力機(jī)制,使用注意力矩陣塊對特征圖進(jìn)行處理,抑制與目標(biāo)分類相關(guān)性低的特征表達(dá),激勵(lì)與目標(biāo)相關(guān)性大的特征表達(dá),從而更有效地利用全局信息對目標(biāo)類型進(jìn)行預(yù)測,抑制無關(guān)信息對分類結(jié)果的干擾,提高目標(biāo)分類的準(zhǔn)確性。此外,本發(fā)明不需要對目標(biāo)周圍的局部圖像做切割操作,因而不存在圖像切割相關(guān)的問題,處理過程更加簡潔,效率更高。