[发明专利]神经网络训练与图像处理方法及装置有效
申请号: | 201910262390.2 | 申请日: | 2019-04-02 |
公开(公告)号: | CN110009090B | 公开(公告)日: | 2022-12-02 |
发明(设计)人: | 曹钰杭;陈恺;吕健勤;林达华 | 申请(专利权)人: | 北京市商汤科技开发有限公司 |
主分类号: | G06N3/02 | 分类号: | G06N3/02;G06N3/08;G06V10/764;G06V10/82 |
代理公司: | 北京林达刘知识产权代理事务所(普通合伙) 11277 | 代理人: | 刘新宇 |
地址: | 100084 北京市海淀区中*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 神经网络 训练 图像 处理 方法 装置 | ||
1.一种神经网络训练方法,其特征在于,包括:
通过目标检测网络对样本图像进行目标检测处理,获得所述样本图像中至少一个对象的多个预测区域,所述样本图像的标注信息包括所述至少一个对象的位置标注信息及至少一个对象的类别标注信息;
基于所述至少一个对象的位置标注信息,确定每个所述预测区域与每个所述预测区域对应的标注区域之间的交并比;
根据所述至少一个对象的类别标注信息以及多个预测区域中每个预测区域的交并比,确定所述多个预测区域的重要性分数;
根据所述多个预测区域的重要性分数以及所述样本图像的标注信息,训练所述目标检测网络;
其中,根据所述至少一个对象的类别标注信息以及多个预测区域中每个预测区域的交并比,确定所述多个预测区域的重要性分数,包括:
根据所述至少一个对象的类别标注信息以及所述多个预测区域中每个预测区域的交并比,对多个预测区域进行排序,获得所述至少一个对象所属的至少一个类别中每个类别对应的预测区域序列;
根据所述多个预测区域中每个预测区域在所述预测区域序列中的排序位置,确定所述每个预测区域的重要性分数;
其中,所述根据所述至少一个对象的类别标注信息以及所述多个预测区域中每个预测区域的交并比,对多个预测区域进行排序,获得所述至少一个对象所属的至少一个类别中每个类别对应的预测区域序列,包括:
根据所述至少一个对象中每个对象的至少两个预测区域的交并比,对所述每个对象的至少两个预测区域进行排序,获得所述每个对象对应的第一序列;
基于所述至少一个对象的类别标注信息,对属于相同类别的至少一个对象对应的第一序列中排序位置相同的预测区域进行排序,获得与所述第一序列中的至少一个排序位置中每个排序位置对应的第二序列;
按照所述第一序列中的排序位置,对所述至少一个排序位置对应的第二序列进行排序,获得所述类别对应的预测区域序列。
2.根据权利要求1所述的方法,其特征在于,所述预测区域序列中的多个预测区域的重要性随着排序位置逐渐下降。
3.根据权利要求1所述的方法,其特征在于,根据所述多个预测区域的重要性分数以及所述样本图像的标注信息,训练所述目标检测网络,包括:
根据所述多个预测区域的重要性分数,确定所述多个预测区域中的每个预测区域的第一权重;
根据所述多个预测区域中每个预测区域的第一权重和所述标注信息,确定所述目标检测网络的分类损失;
根据所述分类损失,调整所述目标检测网络的网络参数。
4.根据权利要求3所述的方法,其特征在于,根据所述多个预测区域中每个预测区域的第一权重和所述标注信息,确定所述目标检测网络的分类损失,包括:
根据所述多个预测区域中每个预测区域的类别预测结果和所述每个预测区域的类别标注信息,确定所述每个预测区域对应的分类损失;
根据所述多个预测区域的第一权重以及所述多个预测区域的分类损失,确定所述多个预测区域中每个预测区域对应的第二权重;
根据所述多个预测区域中每个预测区域的第二权重以及所述多个预测区域的分类损失,确定所述目标检测网络的分类损失。
5.根据权利要求3所述的方法,其特征在于,根据所述多个预测区域的重要性分数以及所述样本图像的标注信息,训练所述目标检测网络,还包括:
根据所述至少一个对象的类别标注信息,确定所述至少一个对象中每个对象的目标类别;
基于所述多个预测区域中每个预测区域在所述每个预测区域对应的目标类别上的类别预测结果,确定所述目标检测网络的回归损失;
基于所述回归损失,调整所述目标检测网络的网络参数。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京市商汤科技开发有限公司,未经北京市商汤科技开发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910262390.2/1.html,转载请声明来源钻瓜专利网。
- 彩色图像和单色图像的图像处理
- 图像编码/图像解码方法以及图像编码/图像解码装置
- 图像处理装置、图像形成装置、图像读取装置、图像处理方法
- 图像解密方法、图像加密方法、图像解密装置、图像加密装置、图像解密程序以及图像加密程序
- 图像解密方法、图像加密方法、图像解密装置、图像加密装置、图像解密程序以及图像加密程序
- 图像编码方法、图像解码方法、图像编码装置、图像解码装置、图像编码程序以及图像解码程序
- 图像编码方法、图像解码方法、图像编码装置、图像解码装置、图像编码程序、以及图像解码程序
- 图像形成设备、图像形成系统和图像形成方法
- 图像编码装置、图像编码方法、图像编码程序、图像解码装置、图像解码方法及图像解码程序
- 图像编码装置、图像编码方法、图像编码程序、图像解码装置、图像解码方法及图像解码程序