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

基本信息

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

摘要

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