基于結(jié)構(gòu)化實(shí)例圖的知識蒸餾方法、裝置、設(shè)備和介質(zhì)

基本信息

申請?zhí)?/td> CN202110551061.7 申請日 -
公開（公告）號 CN113255915A 公開（公告）日 2021-08-13
申請公布號 CN113255915A 申請公布日 2021-08-13
分類號 G06N5/02（2006.01）I;G06K9/62（2006.01）I 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 陳亦新;陳鵬光;賈佳亞;沈小勇;呂江波 申請（專利權(quán)）人 深圳思謀信息科技有限公司
代理機(jī)構(gòu) 廣州華進(jìn)聯(lián)合專利商標(biāo)代理有限公司 代理人 潘思延
地址 518051廣東省深圳市南山區(qū)高新南九道45號三航科技大廈22樓
法律狀態(tài) -

摘要

摘要 本申請涉及一種基于結(jié)構(gòu)化實(shí)例圖的知識蒸餾方法、裝置、計(jì)算機(jī)設(shè)備和存儲介質(zhì)。本申請能夠使得教師網(wǎng)絡(luò)中的前景特征與背景特征都能夠被有效地遷移到學(xué)生網(wǎng)絡(luò)中，進(jìn)一步提高了目標(biāo)檢測的準(zhǔn)確度。該方法包括：獲取訓(xùn)練圖像，分別將訓(xùn)練圖像輸入至教師網(wǎng)絡(luò)學(xué)生網(wǎng)絡(luò)，得到第一特征圖和第二特征圖；再分別將第一特征圖和第二特征圖輸入?yún)^(qū)域候選網(wǎng)絡(luò)中，得到包含邊界框的第一待檢測圖和第二待檢測圖；基于各邊界框中的對象實(shí)例進(jìn)行編碼，得到第一結(jié)構(gòu)圖和第二結(jié)構(gòu)圖；基于第一結(jié)構(gòu)圖和第二結(jié)構(gòu)圖，得到蒸餾損失部分；根據(jù)第二待檢測圖的檢測結(jié)果與真實(shí)標(biāo)簽之間的距離得到基礎(chǔ)損失部分；基于蒸餾損失部分和基礎(chǔ)損失部分，對學(xué)生網(wǎng)絡(luò)進(jìn)行訓(xùn)練。