图像分类模型训练方法、装置、计算机设备和存储介质

基本信息

申请号 CN202110516064.7 申请日 -
公开(公告)号 CN113222014A 公开(公告)日 2021-08-06
申请公布号 CN113222014A 申请公布日 2021-08-06
分类号 G06K9/62(2006.01)I;G06K9/36(2006.01)I 分类 计算;推算;计数;
发明人 黄仕嘉;刘枢;贾佳亚;沈小勇;吕江波 申请(专利权)人 深圳思谋信息科技有限公司
代理机构 广州华进联合专利商标代理有限公司 代理人 蔡抒枫
地址 518051广东省深圳市南山区高新南九道45号三航科技大厦22楼
法律状态 -

摘要

摘要 本申请涉及一种图像分类模型训练方法、装置、计算机设备和存储介质,通过获取样本图像数据集,以及待训练的初始模型的备份模型;将样本图像数据集中的数据划分为多个训练批次数据,并分别将各个训练批次数据输入待训练的初始模型进行训练,得到通过各个训练批次数据训练后的初始模型;将训练后的初始模型与备份模型进行融合,得到目标平滑模型;多个训练批次数据还用于对目标平滑模型进行再次训练;目标平滑模型用于对图像进行分类。本申请通过融合实现了图像分类模型训练过程中的平滑处理,基于目标平滑模型进行训练并重复融合步骤,使得模型在训练中能够保持对于旧图像分类任务的识别准确率,提高了图像分类模型的训练效率和准确率。