圖像識別模型訓練方法及裝置

基本信息

申請?zhí)?/td> CN202111338332.7 申請日 -
公開（公告）號 CN114154559A 公開（公告）日 2022-03-08
申請公布號 CN114154559A 申請公布日 2022-03-08
分類號 G06K9/62（2022.01）I;G06V10/764（2022.01）I;G06V10/774（2022.01）I;G06N3/04（2006.01）I;G06N3/08（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 鐘藝豪;蔡銳濤 申請（專利權）人 有米科技股份有限公司
代理機構 廣州三環(huán)專利商標代理有限公司 代理人 江銀會
地址 510006廣東省廣州市番禺區(qū)小谷圍街青藍街26號1701
法律狀態(tài) -

摘要

摘要 本發(fā)明公開了一種圖像識別模型訓練方法及裝置，該方法包括：確定訓練好的第一圖像識別模型；第一圖像識別模型包括有訓練好的第一特征編碼網(wǎng)絡；確定待訓練的第二圖像識別模型；第二圖像識別模型的模型參數(shù)少于第一圖像識別模型的模型參數(shù)；將第二圖像識別模型中的第二特征編碼網(wǎng)絡的網(wǎng)絡參數(shù)，確定為第一特征編碼網(wǎng)絡的網(wǎng)絡參數(shù)；確定損失函數(shù)為第一特征編碼網(wǎng)絡的特征輸出和第二特征編碼網(wǎng)絡的特征輸出之間的差異，對第一圖像識別模型和第二圖像識別模型進行聯(lián)合訓練，得到訓練好的第二圖像識別模型?？梢?，通過本發(fā)明的方案訓練得到的第二圖像識別模型可以保持較小的規(guī)模的同時達到更好的識別效果。