[发明专利]目标检测深度学习模型训练方法以及目标检测方法在审
| 申请号: | 201910202433.8 | 申请日: | 2019-03-11 |
| 公开(公告)号: | CN109978036A | 公开(公告)日: | 2019-07-05 |
| 发明(设计)人: | 蔡恒;庄浩;张继勇 | 申请(专利权)人: | 华瑞新智科技(北京)有限公司 |
| 主分类号: | G06K9/62 | 分类号: | G06K9/62;G06N3/04 |
| 代理公司: | 北京恒泰铭睿知识产权代理有限公司 11642 | 代理人: | 付怀;何平 |
| 地址: | 100081 北京市海淀区中关*** | 国省代码: | 北京;11 |
| 权利要求书: | 查看更多 | 说明书: | 查看更多 |
| 摘要: | |||
| 搜索关键词: | 目标检测 检测 模型训练 实际场景 训练集 学习 场景 图片 背景图片 光照变化 建立模型 实时运行 训练图片 硬件要求 负样本 鲁棒性 误检率 正样本 | ||
1.一种基于深度学习的目标检测中对深度学习模型进行训练的方法,包括,
建立模型训练图片集,其中包括由不包含检测目标的实际场景图片构成的负样本训练集和由包含检测目标的非实际场景图片构成的正样本训练集;
使用所述模型训练图片集以及其中的图片与检测目标相关的信息,对深度学习模型进行训练。
2.根据权利要求1所述的方法,其中建立模型训练图片集包括对图片进行大小调整。
3.根据权利要求1所述的方法,其中调整模型训练图片集中图片的旋转角度、曝光度、饱和度、色调中至少一个来增加训练数据集数量。
4.根据权利要求1所述的方法,其中图片与检测目标相关的信息包括图片是否包含检测目标,对于包含检测目标的图片,信息还包括目标物体类别及边界框坐标。
5.根据权利要求4所述的方法,其中建立模型训练图片集包括,利用Labe l Image工具将图片与检测目标相关的信息标注到图片。
6.根据权利要求1所述的方法,其中使用所述模型训练图片集对深度学习模型进行训练包括前向传播和反向传播。
7.根据权利要求6所述的方法,其中前向传播获得图片每个网格的各类别概率以及每个边框的坐标信息和含有物体的置信度。
8.根据权利要求7所述的方法,其中反向传播包括,根据前向传播获得图片每个网格的各类别概率以及每个边框的坐标信息和含有物体的置信度以及图片的与检测目标相关的信息,通过损失函数计算深度学习模型的损失。
9.根据权利要求8所述的方法,其中反向传播还包括,根据深度学习模型的损失计算敏感度图,然后计算偏置更新需要的梯度和权重更新需要的梯度,最后根据梯度通过梯度下降对偏置和权重进行更新。
10.一种基于深度学习的目标检测方法,其中的深度学习模型是使用根据权利要求1-9任一项所述的方法训练的。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于华瑞新智科技(北京)有限公司,未经华瑞新智科技(北京)有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910202433.8/1.html,转载请声明来源钻瓜专利网。





