图像分类模型训练方法、装置、计算机设备和存储介质
基本信息
申请号 | CN202110516064.7 | 申请日 | - |
公开(公告)号 | CN113222014A | 公开(公告)日 | 2021-08-06 |
申请公布号 | CN113222014A | 申请公布日 | 2021-08-06 |
分类号 | G06K9/62(2006.01)I;G06K9/36(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 黄仕嘉;刘枢;贾佳亚;沈小勇;吕江波 | 申请(专利权)人 | 深圳思谋信息科技有限公司 |
代理机构 | 广州华进联合专利商标代理有限公司 | 代理人 | 蔡抒枫 |
地址 | 518051广东省深圳市南山区高新南九道45号三航科技大厦22楼 | ||
法律状态 | - |
摘要
摘要 | 本申请涉及一种图像分类模型训练方法、装置、计算机设备和存储介质,通过获取样本图像数据集,以及待训练的初始模型的备份模型;将样本图像数据集中的数据划分为多个训练批次数据,并分别将各个训练批次数据输入待训练的初始模型进行训练,得到通过各个训练批次数据训练后的初始模型;将训练后的初始模型与备份模型进行融合,得到目标平滑模型;多个训练批次数据还用于对目标平滑模型进行再次训练;目标平滑模型用于对图像进行分类。本申请通过融合实现了图像分类模型训练过程中的平滑处理,基于目标平滑模型进行训练并重复融合步骤,使得模型在训练中能够保持对于旧图像分类任务的识别准确率,提高了图像分类模型的训练效率和准确率。 |
