模型训练方法、装置、计算机设备和存储介质
基本信息
申请号 | CN202210274888.2 | 申请日 | - |
公开(公告)号 | CN114359563B | 公开(公告)日 | 2022-06-28 |
申请公布号 | CN114359563B | 申请公布日 | 2022-06-28 |
分类号 | G06V10/26(2022.01)I;G06V10/74(2022.01)I;G06V10/774(2022.01)I;G06K9/62(2022.01)I | 分类 | 计算;推算;计数; |
发明人 | 田倬韬;易振彧;刘枢;吕江波;沈小勇 | 申请(专利权)人 | 苏州思谋智能科技有限公司 |
代理机构 | 华进联合专利商标代理有限公司 | 代理人 | - |
地址 | 518051广东省深圳市南山区高新南九道45号三航科技大厦22楼 | ||
法律状态 | - |
摘要
摘要 | 本申请涉及一种模型训练方法、装置、计算机设备、存储介质和计算机程序产品。所述方法包括:获取已训练的教师模型对样本图像中各像素的类别预测结果;根据已训练的教师模型对样本图像中各像素的类别预测结果,得到样本图像中各像素的信息量;在各预设图像类别下,基于样本图像中各像素的信息量、已训练的教师模型对样本图像中各像素的类别预测结果和待训练的学生模型对样本图像中各像素的类别预测结果,得到待训练的学生模型的目标损失函数;根据目标损失函数,对待训练的学生模型进行迭代训练,得到训练完成的学生模型;训练完成的学生模型用于对输入的图像进行语义分割。采用本方法能够提升学生模型整体的预测准确性。 |
