圖像分類模型的訓練方法及圖像分類方法

基本信息

申請?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 分類 計算；推算；計數;
發(fā)明人 劉國清;楊廣;王啟程;鄭偉;張孟華;楊國武 申請（專利權）人 深圳佑駕創(chuàng)新科技有限公司
代理機構 深圳市倡創(chuàng)專利代理事務所（普通合伙） 代理人 羅明玉
地址 518049廣東省深圳市福田區(qū)梅林街道梅都社區(qū)中康路136號深圳新一代產業(yè)園1棟401
法律狀態(tài) -

摘要

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