一種目標檢測模型訓練方法、裝置、電子設(shè)備及存儲介質(zhì)

基本信息

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

摘要

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