[发明专利]对象识别模型的训练方法、对象识别方法及相应的装置有效
| 申请号： | 201910690389.X | 申请日： | 2019-07-29 |
| 公开（公告）号： | CN110414432B | 公开（公告）日： | 2023-05-16 |
| 发明（设计）人： | 蒋忻洋;朱智慧;孙星;郭晓威 | 申请（专利权）人： | 腾讯科技（深圳）有限公司 |
| 主分类号： | G06V10/62 | 分类号： | G06V10/62;G06V40/10;G06V20/40;G06V10/774 |
| 代理公司： | 北京市立方律师事务所 11330 | 代理人： | 张筱宁 |
| 地址： | 518000 广东省深圳*** | 国省代码： | 广东;44 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | 本申请实施例提供了一种对象识别模型的训练方法、对象识别方法及相应的装置，该方法包括：获取训练样本集；构建包括对象识别模型，以及分别与对象识别模型的输出连接的第一分类模块和第二分类模块的待训练模型；将各样本图像输入至待训练模型，根据第一分类模型输出的预测身份信息和样本图像的标注的身份信息确定第一训练损失值，根据第二分类模型输出的预测身份及朝向信息，与样本图像的标注的身份信息及朝向信息，确定第二训练损失值；根据由第一训练损失值和第二训练损失值确定的总损失值对待训练模型的模型参数进行调整，直至总损失值满足预设条件，得到训练好的对象识别模型。基于该方法，提高了将对象识别模型用于对象识别时的识别准确性。 | ||
| 搜索关键词： | 对象 识别 模型 训练 方法 相应 装置 | ||
【主权项】：
1.一种对象识别模型的训练方法，其特征在于，包括：获取训练样本集，所述训练样本集中包括对象的样本图像，所述样本图像标注有图像中的对象的身份标签和朝向标签，所述身份标签用于表征对象的身份信息，所述朝向标签用于表征对象的朝向信息；构建待训练模型，所述待训练模型包括所述对象识别模型，以及分别与所述对象识别模型的输出连接的第一分类模块和第二分类模块，其中，所述第一分类模型用于输出所述样本图像中对象的预测身份信息，所述第二分类模型用于输出所述样本图像中对象的预测身份及朝向信息；将各样本图像输入至所述待训练模型中，根据所述第一分类模型输出的预测身份信息和样本图像的身份标签所对应的身份信息确定第一训练损失值，根据所述第二分类模型输出的预测身份及朝向信息，与样本图像的身份标签和朝向标签所对应的身份信息及朝向信息，确定第二训练损失值；基于所述第一训练损失值和所述第二训练损失值，确定总损失值；根据所述总损失值对所述待训练模型的模型参数进行调整，直至所述总损失值满足预设条件，得到训练好的对象识别模型。
下载完整专利技术内容需要扣除积分，VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于腾讯科技（深圳）有限公司，未经腾讯科技（深圳）有限公司许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/patent/201910690389.X/，转载请声明来源钻瓜专利网。
