圖像分類模型的訓(xùn)練方法及圖像分類方法

基本信息

申請?zhí)?/td> CN202111459523.9 申請日 -
公開（公告）號 CN113869464B 公開（公告）日 2022-03-18
申請公布號 CN113869464B 申請公布日 2022-03-18
分類號 G06K9/62（2022.01）I;G06N3/08（2006.01）I;G06N3/04（2006.01）I;G06V10/774（2022.01）I;G06V10/82（2022.01）I;G06V10/764（2022.01）I 分類 計算；推算；計數(shù);
發(fā)明人 劉國清;楊廣;王啟程;鄭偉;張孟華;楊國武 申請（專利權(quán)）人 深圳佑駕創(chuàng)新科技有限公司
代理機構(gòu) 深圳市倡創(chuàng)專利代理事務(wù)所（普通合伙） 代理人 羅明玉
地址 518049廣東省深圳市福田區(qū)梅林街道梅都社區(qū)中康路136號深圳新一代產(chǎn)業(yè)園1棟401
法律狀態(tài) -

摘要

摘要 本發(fā)明提供一種圖像分類模型的訓(xùn)練方法，包括：將第一圖像劃分為若干子集，第一圖像為有標簽圖像；構(gòu)建若干合集，每一合集包括一一對應(yīng)的驗證集和訓(xùn)練集，每一合集中，驗證集為若干子集中的一個子集，訓(xùn)練集包括第二圖像集和若干子集中的其余子集，第二圖像集中的第二圖像為無標簽圖像；將每一訓(xùn)練集分別輸入若干對學生網(wǎng)絡(luò)和教師網(wǎng)絡(luò)，并獲取相應(yīng)的輸出結(jié)果；根據(jù)輸出結(jié)果更新學生網(wǎng)絡(luò)的第一參數(shù)和教師網(wǎng)絡(luò)的第二參數(shù)；將相對應(yīng)的驗證集中的第一圖像輸入學生網(wǎng)絡(luò)并獲取相應(yīng)的驗證結(jié)果，并根據(jù)驗證結(jié)果選取最優(yōu)學生網(wǎng)絡(luò)；以及將若干最優(yōu)學生網(wǎng)絡(luò)作為圖像分類模型。本發(fā)明技術(shù)方案有效解決了有標簽圖像數(shù)據(jù)量小導(dǎo)致圖像分類模型準確度不高的問題。