[发明专利]一种目标检测网络训练方法、装置及电子设备有效
申请号: | 201910277616.6 | 申请日: | 2019-04-08 |
公开(公告)号: | CN109977913B | 公开(公告)日: | 2021-11-05 |
发明(设计)人: | 王冬冬 | 申请(专利权)人: | 北京奇艺世纪科技有限公司 |
主分类号: | G06K9/00 | 分类号: | G06K9/00;G06K9/32;G06N3/04 |
代理公司: | 北京柏杉松知识产权代理事务所(普通合伙) 11413 | 代理人: | 丁芸;项京 |
地址: | 100080 北京市海淀区*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | 本发明实施例提供了一种目标检测网络训练方法、装置及电子设备,其中,该方法包括:获取携带目标的样本和未携带目标的样本,将携带目标的样本输入至第一SSD网络,得到第一SSD网络中的第一输出层输出的类别损失和位置损失;并将未携带目标的样本输入至第二SSD网络,得到第二SSD网络中的第二输出层输出的类别损失,其中,第二SSD网络与第一SSD网络具有相同的网络参数;对第一输出层输出的类别损失和位置损失以及第二输出层输出的类别损失求和,得到总损失,并基于总损失更新第一SSD网络中的网络参数和第二SSD网络中的网络参数。可以降低第一SSD网络从非目标位置检测出目标的可能性,从而提高训练后的第一SSD网络检测的准确度。 | ||
搜索关键词: | 一种 目标 检测 网络 训练 方法 装置 电子设备 | ||
【主权项】:
1.一种目标检测网络训练方法,其特征在于,所述方法包括:获取携带目标的样本和未携带所述目标的样本,其中,所述携带目标的样本中标注有所述目标的类别和所述目标的位置;将所述携带目标的样本输入至第一单次多框检测器SSD网络,得到所述第一SSD网络中的第一输出层输出的类别损失和位置损失;并将所述未携带所述目标的样本输入至第二SSD网络,得到所述第二SSD网络中的第二输出层输出的类别损失,其中,所述第二SSD网络与所述第一SSD网络具有相同的网络参数;对所述第一输出层输出的类别损失和位置损失以及所述第二输出层输出的类别损失求和,得到总损失,并基于所述总损失更新所述第一SSD网络中的网络参数和所述第二SSD网络中的网络参数。
下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京奇艺世纪科技有限公司,未经北京奇艺世纪科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/patent/201910277616.6/,转载请声明来源钻瓜专利网。