图像分类模型的训练方法及计算机可读存储介质
基本信息
| 申请号 | 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 | 分类 | 计算；推算；计数; |
| 发明人 | 刘国清;杨广;王启程;郑伟;张孟华;杨国武 | 申请（专利权）人 | 深圳佑驾创新科技有限公司 |
| 代理机构 | 深圳市倡创专利代理事务所（普通合伙） | 代理人 | 罗明玉 |
| 地址 | 518049广东省深圳市福田区梅林街道梅都社区中康路136号深圳新一代产业园1栋401 | ||
| 法律状态 | - | ||
摘要
| 摘要 | 本发明提供了一种图像分类模型的训练方法，包括：根据第一图像训练初始分类模型以得到中间分类模型，其中，第一图像为有标签图像；根据中间分类模型对第二图像进行聚类处理以得到若干聚类簇，其中，第二图像为无标签图像；计算每一聚类簇的中心点；根据聚类簇的中心点计算相应的伪标签；以及根据第一图像、第二图像、中心点以及伪标签训练中间分类模型以得到图像分类模型。此外，本发明还提供了一种计算机可读存储介质。本发明技术方案有效解决了有标签图像数据量小导致图像分类模型准确度不高的问题。 |
