[发明专利]物体检测模型训练方法及装置在审
| 申请号： | 201910659672.6 | 申请日： | 2019-07-19 |
| 公开（公告）号： | CN112241675A | 公开（公告）日： | 2021-01-19 |
| 发明（设计）人： | 周定富;方进;宋希彬;官晨晔;杨睿刚 | 申请（专利权）人： | 百度在线网络技术（北京）有限公司 |
| 主分类号： | G06K9/00 | 分类号： | G06K9/00;G06K9/62;G06N3/08 |
| 代理公司： | 北京市铸成律师事务所 11313 | 代理人： | 杨瑾瑾;林蕾 |
| 地址： | 100085 北京市*** | 国省代码： | 北京;11 |
| 权利要求书： | 查看更多 | 说明书： | 查看更多 |
| 摘要： | |||
| 搜索关键词： | 物体 检测 模型 训练 方法 装置 | ||
1.一种物体检测模型训练方法，其特征在于，包括：
根据检测目标的标注数据确定物体真值包围框；
将所述检测目标输入物体检测模型，并根据所述物体检测模型输出的检测数据得到物体检测包围框；
根据所述物体真值包围框和所述物体检测包围框，确定所述物体真值包围框和所述物体检测包围框的交并比；
根据所述交并比，确定损失函数值；
根据所述损失函数值，优化所述物体检测模型。
2.根据权利要求1所述的方法，其特征在于，在所述物体真值包围框和所述物体检测包围框之间存在交叠的情况下，根据所述交并比，确定损失函数值，包括：
根据下述公式，确定损失函数值：
L＝1-IoU；
其中L为所述损失函数值，IoU为所述交并比。
3.根据权利要求1所述的方法，其特征在于，在所述物体真值包围框和所述物体检测包围框之间不存在交叠的情况下，根据所述交并比，确定损失函数值，包括：
根据下述公式，确定损失函数值：
L＝1-GIoU；
其中，
其中，AreaC为包含所述物体真值包围框和所述物体检测包围框的最小包围框的面积，AreaU为所述物体真值包围框和所述物体检测包围框的并集的面积，IoU为所述交并比。
4.根据权利要求1所述的方法，其特征在于，根据所述损失函数值，优化所述物体检测模型，包括：
根据所述损失函数值，进行反向传播计算，获得反向传播计算梯度值；
根据所述反向传播计算梯度值，优化所述物体检测模型的参数。
5.根据权利要求1所述的方法，其特征在于，根据所述物体真值包围框和所述物体检测包围框，确定所述物体真值包围框和所述物体检测包围框的交并比，包括：
确定所述物体检测包围框和所述物体真值包围框的交集和并集；
根据所述交集和所述并集，确定所述交并比。
6.根据权利要求1-5中任意一项所述的方法，其特征在于，所述物体真值包围框的任意边与所述物体检测包围框的任意边处于非平行状态。
7.一种物体检测模型训练装置，其特征在于，包括：
真值模块：用于根据检测目标的标注数据确定物体真值包围框；
检测模块：用于将所述检测目标输入物体检测模型，并根据所述物体检测模型输出的检测数据得到物体检测包围框；
交并比计算模块：用于根据所述物体真值包围框和所述物体检测包围框，确定所述物体真值包围框和所述物体检测包围框的交并比；
损失函数计算模块：用于根据所述交并比，确定损失函数值；
优化模块：用于根据所述损失函数值，优化所述物体检测模型。
8.根据权利要求7所述的装置，其特征在于，在所述物体真值包围框和所述物体检测包围框之间存在交叠的情况下，所述损失函数计算模块用于：
根据下述公式，确定损失函数值：
L＝1-IoU；
其中L为所述损失函数值，IoU为所述交并比。
9.根据权利要求7所述的装置，其特征在于，在所述物体真值包围框和所述物体检测包围框之间不存在交叠的情况下，所述损失函数计算模块用于：
根据下述公式，确定损失函数值：
L＝1-GIoU；
其中，
其中，AreaC为包含所述物体真值包围框和所述物体检测包围框的最小包围框的面积，AreaU为所述物体真值包围框和所述物体检测包围框的并集的面积，IoU为所述交并比。
10.根据权利要求7所述的装置，其特征在于，所述优化模块包括：
反向传播计算单元：用于根据所述损失函数值，进行反向传播计算，获得反向传播计算梯度值；
反向处理单元：用于根据所述反向传播计算梯度值，优化所述物体检测模型的参数。
该专利技术资料仅供研究查看技术是否侵权等信息，商用须获得专利权人授权。该专利全部权利属于百度在线网络技术（北京）有限公司，未经百度在线网络技术（北京）有限公司许可，擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作，请联系【客服】
本文链接：http://www.vipzhuanli.com/pat/books/201910659672.6/1.html，转载请声明来源钻瓜专利网。
