分类模型的训练方法、装置、电子设备以及存储介质

基本信息

申请号 CN202110625599.8 申请日 -
公开(公告)号 CN113239883A 公开(公告)日 2021-08-10
申请公布号 CN113239883A 申请公布日 2021-08-10
分类号 G06K9/00(2006.01)I;G06K9/62(2006.01)I;G06N3/08(2006.01)I 分类 计算;推算;计数;
发明人 陈文敏;王欢;冯歆鹏;周骥 申请(专利权)人 上海肇观电子科技有限公司
代理机构 北京品源专利代理有限公司 代理人 孟金喆
地址 201203上海市浦东新区自由贸易试验区龙东大道3000号7幢501室
法律状态 -

摘要

摘要 本发明公开了一种分类模型的训练方法、装置、电子设备以及存储介质,属于人工智能技术领域。该方法包括:通过初始模型的目标分类分支,对样本图像进行处理得到分类损失值;所述目标分类分支用于预测样本图像中目标对象的目标分类结果;通过初始模型的样本识别分支,对所述样本图像进行处理得到识别损失值;所述样本识别分支用于预测样本图像中目标对象的样本识别结果;通过初始模型的权重调节分支,对所述样本图像进行处理,得到对所述目标分类分支和所述样本识别分支进行权重调节的损失权重;根据所述分类损失值、所述识别损失值和所述损失权重,对所述初始模型进行训练,得到目标分类模型。上述技术方案,提高了模型的分类准确率。