圖像識別模型訓(xùn)練方法、裝置、設(shè)備及介質(zhì)

基本信息

申請?zhí)?/td> CN202110701766.2 申請日 -
公開（公告）號 CN113505820A 公開（公告）日 2021-10-15
申請公布號 CN113505820A 申請公布日 2021-10-15
分類號 G06K9/62（2006.01）I;G06K9/46（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 宋麗;劉利 申請（專利權(quán)）人 北京閱視智能技術(shù)有限責(zé)任公司
代理機構(gòu) 北京東方億思知識產(chǎn)權(quán)代理有限責(zé)任公司 代理人 趙秀芹
地址 100085北京市海淀區(qū)上地創(chuàng)業(yè)中路28號1號樓2層201
法律狀態(tài) -

摘要

摘要 本申請公開了一種圖像識別模型訓(xùn)練方法、裝置、設(shè)備及介質(zhì)。其中，圖像識別模型訓(xùn)練方法，包括：建立初始圖像識別模型；初始圖像識別模型中具有損失函數(shù)；初始圖像識別模型包括特征提取網(wǎng)絡(luò)與分類網(wǎng)絡(luò)，其中，特征提取網(wǎng)絡(luò)用于接收攜帶標(biāo)簽的樣本圖像，并輸出特征圖；分類網(wǎng)絡(luò)用于接收特征圖，并根據(jù)特征圖中的K個目標(biāo)特征點輸出初始識別結(jié)果；在使用訓(xùn)練樣本集對初始圖像識別模型進(jìn)行訓(xùn)練的過程中，按照預(yù)設(shè)衰減規(guī)則更新K的值，按照損失函數(shù)的損失值更新初始圖像識別模型的網(wǎng)絡(luò)參數(shù)；并在損失函數(shù)的損失值小于損失閾值的情況下，獲得訓(xùn)練完成的圖像識別模型。本申請實施例可以有效提高訓(xùn)練完成的圖像識別模型的識別能力。