[发明专利]目标检测及目标检测网络的训练方法、装置及设备有效
申请号: | 201910563005.8 | 申请日: | 2019-06-26 |
公开(公告)号: | CN110298298B | 公开(公告)日: | 2022-03-08 |
发明(设计)人: | 李聪 | 申请(专利权)人: | 北京市商汤科技开发有限公司 |
主分类号: | G06V20/40 | 分类号: | G06V20/40;G06V10/26;G06T7/194 |
代理公司: | 北京博思佳知识产权代理有限公司 11415 | 代理人: | 靳玫 |
地址: | 100080 北京市海淀区北*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 目标 检测 网络 训练 方法 装置 设备 | ||
1.一种目标检测方法,其特征在于,所述方法包括:
获得输入图像的特征数据;
根据所述特征数据,确定所述输入图像的多个候选边界框;
根据所述特征数据,获得所述输入图像的前景分割结果,其中,前景分割结果包含指示所述输入图像的多个像素中每个像素是否属于前景的指示信息;
根据所述多个候选边界框与所述前景分割结果,得到所述输入图像的目标检测结果,包括:
将所述多个候选边界框中与所述前景分割结果对应的前景图像区域之间的重叠区域在整个候选边界框中所占的比例大于第一阈值的候选边界框作为目标边界框;
基于所述目标边界框,得到所述输入图像的目标检测结果。
2.根据权利要求1所述的方法,其特征在于,所述目标边界框包括第一边界框和第二边界框,所述基于所述目标边界框,得到所述输入图像的目标检测结果,包括:
基于所述第一边界框和所述第二边界框之间的夹角,确定所述第一边界框和所述第二边界框的重叠参数;
基于所述第一边界框和所述第二边界框的重叠参数,确定所述第一边界框和所述第二边界框所对应的目标对象位置。
3.根据权利要求2所述的方法,其特征在于,所述基于所述第一边界框和所述第二边界框之间的夹角,确定所述第一边界框和所述第二边界框的重叠参数,包括:
根据所述第一边界框和所述第二边界框之间的夹角,获得角度因子;
根据所述第一边界框和所述第二边界框之间的交并比和所述角度因子,获得所述重叠参数。
4.根据权利要求3所述的方法,其特征在于,所述重叠参数为所述交并比与所述角度因子的乘积,其中,所述角度因子随着所述第一边界框和所述第二边界框之间的角度的增大而增大。
5.根据权利要求4所述的方法,其特征在于,在所述交并比保持一定的条件下,所述重叠参数随着所述第一边界框和所述第二边界框之间的角度的增大而增大。
6.根据权利要求2所述的方法,其特征在于,在所述重叠参数大于第二阈值的情况下,将所述第一边界框和所述边界框中的其中一个边界框作为目标对象位置。
7.根据权利要求6所述的方法,其特征在于,所述将所述第一边界框和所述边界框中的其中一个边界框作为目标对象位置,包括:
确定所述第一边界框与所述前景分割结果对应的前景图像区域之间的重叠参数和所述第二边界框与所述前景图像区域之间的重叠参数;
将所述第一边界框和所述第二边界框中重叠参数较大的边界框作为目标对象位置。
8.根据权利要求2所述的方法,其特征在于,在所述重叠参数小于或等于第二阈值的情况下,将所述第一边界框和第二边界框均作为目标对象位置。
9.根据权利要求1所述的方法,其特征在于,待检测的目标对象的长宽比大于特定数值。
10.一种目标检测网络的训练方法,其特征在于,所述目标检测网络包括特征提取网络、目标预测网络和前景分割网络,所述方法包括:
通过所述特征提取网络对样本图像进行特征提取处理,获得所述样本图像的特征数据;
根据所述特征数据,通过所述目标预测网络获得多个样本候选边界框;
根据所述特征数据,通过所述前景分割网络获得所述样本图像的样本前景分割结果,其中,所述样本前景分割结果包含指示所述样本图像的多个像素点中每个像素点是否属于前景的指示信息;
根据所述多个样本候选边界框和所述样本前景分割结果以及所述样本图像的标注信息,确定网络损失值;
基于所述网络损失值,对所述目标检测网络的网络参数进行调整,
在确定所述网络损失值的过程中,所述样本候选边界框的宽度所对应的权重高于所述样本候选边界框的长度所对应的权重。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京市商汤科技开发有限公司,未经北京市商汤科技开发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910563005.8/1.html,转载请声明来源钻瓜专利网。
- 上一篇:火焰识别方法和装置
- 下一篇:一种基于等高航线的无人机边坡植被分类方法