一種目標(biāo)檢測(cè)模型訓(xùn)練方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)

基本信息

申請(qǐng)?zhí)?/td> CN202110186629.X 申請(qǐng)日 -
公開（公告）號(hào) CN112560999B 公開（公告）日 2021-06-04
申請(qǐng)公布號(hào) CN112560999B 申請(qǐng)公布日 2021-06-04
分類號(hào) G06K9/62;G06N3/04;G06N3/08 分類 計(jì)算；推算；計(jì)數(shù);
發(fā)明人 王樹;羅振波;吉翔 申請(qǐng)（專利權(quán)）人 成都睿沿科技有限公司
代理機(jī)構(gòu) 北京超凡宏宇專利代理事務(wù)所（特殊普通合伙） 代理人 蔣姍
地址 610000 四川省成都市高新區(qū)中國（四川）自由貿(mào)易試驗(yàn)區(qū)益州大道中段1800號(hào)4棟6層601號(hào)
法律狀態(tài) -

摘要

摘要 本申請(qǐng)?zhí)峁┮环N目標(biāo)檢測(cè)模型訓(xùn)練方法、裝置、電子設(shè)備及存儲(chǔ)介質(zhì)，該方法包括：獲得多個(gè)樣本圖像和多個(gè)類別熱力圖像；以多個(gè)樣本圖像為訓(xùn)練數(shù)據(jù)，以多個(gè)類別熱力圖像為數(shù)據(jù)標(biāo)簽，對(duì)目標(biāo)檢測(cè)模型中的特征提取網(wǎng)絡(luò)和分類注意力網(wǎng)絡(luò)進(jìn)行訓(xùn)練，并以標(biāo)注圖像為數(shù)據(jù)標(biāo)簽，對(duì)目標(biāo)檢測(cè)模型中的特征提取網(wǎng)絡(luò)和檢測(cè)網(wǎng)絡(luò)進(jìn)行訓(xùn)練，通過聯(lián)合訓(xùn)練的方式獲得訓(xùn)練后的目標(biāo)檢測(cè)模型。通過在目標(biāo)檢測(cè)模型加入了能夠提取和有效利用不同類別間的差異化特征的分類注意力網(wǎng)絡(luò)，從而引導(dǎo)模型去定位待檢測(cè)圖像中的具有重要影響的關(guān)鍵區(qū)域，讓模型更好地關(guān)注關(guān)鍵區(qū)域從而學(xué)習(xí)到易混淆類別之間的差異性，提高了對(duì)圖像進(jìn)行目標(biāo)檢測(cè)的正確率。