图像识别模型训练方法及装置

基本信息

申请号 CN202111338332.7 申请日 -
公开(公告)号 CN114154559A 公开(公告)日 2022-03-08
申请公布号 CN114154559A 申请公布日 2022-03-08
分类号 G06K9/62(2022.01)I;G06V10/764(2022.01)I;G06V10/774(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 钟艺豪;蔡锐涛 申请(专利权)人 有米科技股份有限公司
代理机构 广州三环专利商标代理有限公司 代理人 江银会
地址 510006广东省广州市番禺区小谷围街青蓝街26号1701
法律状态 -

摘要

摘要 本发明公开了一种图像识别模型训练方法及装置,该方法包括:确定训练好的第一图像识别模型;第一图像识别模型包括有训练好的第一特征编码网络;确定待训练的第二图像识别模型;第二图像识别模型的模型参数少于第一图像识别模型的模型参数;将第二图像识别模型中的第二特征编码网络的网络参数,确定为第一特征编码网络的网络参数;确定损失函数为第一特征编码网络的特征输出和第二特征编码网络的特征输出之间的差异,对第一图像识别模型和第二图像识别模型进行联合训练,得到训练好的第二图像识别模型。可见,通过本发明的方案训练得到的第二图像识别模型可以保持较小的规模的同时达到更好的识别效果。