模型訓練方法、裝置、計算機設備和存儲介質

基本信息

申請?zhí)?/td> CN202210274888.2 申請日 -
公開（公告）號 CN114359563B 公開（公告）日 2022-06-28
申請公布號 CN114359563B 申請公布日 2022-06-28
分類號 G06V10/26（2022.01）I;G06V10/74（2022.01）I;G06V10/774（2022.01）I;G06K9/62（2022.01）I 分類 計算；推算；計數(shù);
發(fā)明人 田倬韜;易振彧;劉樞;呂江波;沈小勇 申請（專利權）人 蘇州思謀智能科技有限公司
代理機構 華進聯(lián)合專利商標代理有限公司 代理人 -
地址 518051廣東省深圳市南山區(qū)高新南九道45號三航科技大廈22樓
法律狀態(tài) -

摘要

摘要 本申請涉及一種模型訓練方法、裝置、計算機設備、存儲介質和計算機程序產(chǎn)品。所述方法包括：獲取已訓練的教師模型對樣本圖像中各像素的類別預測結果；根據(jù)已訓練的教師模型對樣本圖像中各像素的類別預測結果，得到樣本圖像中各像素的信息量；在各預設圖像類別下，基于樣本圖像中各像素的信息量、已訓練的教師模型對樣本圖像中各像素的類別預測結果和待訓練的學生模型對樣本圖像中各像素的類別預測結果，得到待訓練的學生模型的目標損失函數(shù)；根據(jù)目標損失函數(shù)，對待訓練的學生模型進行迭代訓練，得到訓練完成的學生模型；訓練完成的學生模型用于對輸入的圖像進行語義分割。采用本方法能夠提升學生模型整體的預測準確性。