圖像分類模型的訓練方法及計算機可讀存儲介質

基本信息

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

摘要

摘要 本發(fā)明提供了一種圖像分類模型的訓練方法,包括:根據(jù)第一圖像訓練初始分類模型以得到中間分類模型,其中,第一圖像為有標簽圖像;根據(jù)中間分類模型對第二圖像進行聚類處理以得到若干聚類簇,其中,第二圖像為無標簽圖像;計算每一聚類簇的中心點;根據(jù)聚類簇的中心點計算相應的偽標簽;以及根據(jù)第一圖像、第二圖像、中心點以及偽標簽訓練中間分類模型以得到圖像分類模型。此外,本發(fā)明還提供了一種計算機可讀存儲介質。本發(fā)明技術方案有效解決了有標簽圖像數(shù)據(jù)量小導致圖像分類模型準確度不高的問題。