圖像分類模型訓練方法、裝置、計算機設(shè)備和存儲介質(zhì)

基本信息

申請?zhí)?/td> CN202110516064.7 申請日 -
公開（公告）號 CN113222014A 公開（公告）日 2021-08-06
申請公布號 CN113222014A 申請公布日 2021-08-06
分類號 G06K9/62（2006.01）I;G06K9/36（2006.01）I 分類 計算；推算；計數(shù);
發(fā)明人 黃仕嘉;劉樞;賈佳亞;沈小勇;呂江波 申請（專利權(quán)）人 深圳思謀信息科技有限公司
代理機構(gòu) 廣州華進聯(lián)合專利商標代理有限公司 代理人 蔡抒楓
地址 518051廣東省深圳市南山區(qū)高新南九道45號三航科技大廈22樓
法律狀態(tài) -

摘要

摘要 本申請涉及一種圖像分類模型訓練方法、裝置、計算機設(shè)備和存儲介質(zhì)，通過獲取樣本圖像數(shù)據(jù)集，以及待訓練的初始模型的備份模型；將樣本圖像數(shù)據(jù)集中的數(shù)據(jù)劃分為多個訓練批次數(shù)據(jù)，并分別將各個訓練批次數(shù)據(jù)輸入待訓練的初始模型進行訓練，得到通過各個訓練批次數(shù)據(jù)訓練后的初始模型；將訓練后的初始模型與備份模型進行融合，得到目標平滑模型；多個訓練批次數(shù)據(jù)還用于對目標平滑模型進行再次訓練；目標平滑模型用于對圖像進行分類。本申請通過融合實現(xiàn)了圖像分類模型訓練過程中的平滑處理，基于目標平滑模型進行訓練并重復融合步驟，使得模型在訓練中能夠保持對于舊圖像分類任務的識別準確率，提高了圖像分類模型的訓練效率和準確率。