[发明专利]目标检测模型的训练方法及装置、目标检测方法及装置在审

专利信息
申请号: 202210520731.3 申请日: 2022-05-13
公开(公告)号: CN114782680A 公开(公告)日: 2022-07-22
发明(设计)人: 赵梦傲;张骞;黄畅 申请(专利权)人: 北京地平线信息技术有限公司
主分类号: G06V10/25 分类号: G06V10/25;G06V10/40;G06V10/774;G06V10/82;G06K9/62;G06N3/04;G06N3/08
代理公司: 北京思源智汇知识产权代理有限公司 11657 代理人: 靳涛涛
地址: 100094 北京市海淀*** 国省代码: 北京;11
权利要求书: 查看更多 说明书: 查看更多
摘要:
搜索关键词: 目标 检测 模型 训练 方法 装置
【权利要求书】:

1.一种目标检测模型的训练方法,包括:

利用待训练的目标检测模型包括的由级联的至少两层子网络组成的主干网络,对样本图像进行逐级特征提取,得到特征数据,其中,所述至少两层子网络按照数据处理量的大小顺序排列;

利用所述待训练的目标检测模型包括的检测网络,对所述特征数据进行检测,得到所述样本图像中的目标对象的检测结果;

基于预设的损失函数,确定表示所述检测结果和预先对所述目标对象标注的标注信息之间的误差的损失值;

基于所述损失值,调整所述待训练的目标检测模型的参数。

2.根据权利要求1所述的方法,其中,所述利用待训练的目标检测模型包括的由级联的至少两层子网络组成的主干网络,对样本图像进行逐级特征提取,得到特征数据,包括:

确定所述至少两层子网络中当前运行的目标子网络在所述主干网络中的位置;

由所述目标子网络包括的与所述位置对应的数据处理单元,对输入所述目标子网络的数据进行处理,得到所述目标子网络对应的子特征数据;

基于所述子特征数据,生成所述主干网络对应的特征数据。

3.根据权利要求2所述的方法,其中,所述由所述目标子网络包括的与所述位置对应的数据处理单元,对输入所述目标子网络的数据进行处理,得到所述目标子网络对应的子特征数据,包括:

响应于确定所述目标子网络包括的与所述位置对应的数据处理单元包括下采样单元,利用所述下采样单元包括的至少一条数据处理分支,对输入所述下采样单元的数据进行下采样处理,得到至少一组待合并数据;

基于所述至少一组待合并数据,生成下采样特征数据;

基于所述下采样特征数据,确定所述目标子网络对应的子特征数据。

4.根据权利要求3所述的方法,其中,所述利用所述下采样单元包括的至少一条数据处理分支,对输入所述下采样单元的数据进行下采样处理,得到至少一组待合并数据,包括:

利用所述下采样单元中的至少一个组卷积子单元和至少一个逐点卷积子单元构成的至少一条数据处理分支,对输入所述下采样单元的数据进行下采样处理,得到至少一组待合并数据。

5.根据权利要求2所述的方法,其中,所述由所述目标子网络包括的与所述位置对应的数据处理单元,对输入所述目标子网络的数据进行处理,得到所述目标子网络对应的子特征数据,包括:

响应于确定所述目标子网络包括的与所述位置对应的数据处理单元包括特征提取单元,对输入所述特征提取单元的数据进特征提取处理,得到中间特征数据;

基于所述中间特征数据,确定所述目标子网络对应的子特征数据。

6.根据权利要求5所述的方法,其中,所述对输入所述特征提取单元的数据进特征提取处理,得到中间特征数据,包括:

利用所述特征提取单元中的至少一个组卷积子单元和至少一个逐点卷积子单元,对输入所述特征提取单元的数据进行特征提取处理,并利用所述特征提取单元中的捷径分支对数据进行跨层处理,得到中间特征数据。

7.根据权利要求5所述的方法,其中,所述响应于确定所述目标子网络包括的与所述位置对应的数据处理单元包括特征提取单元,对输入所述特征提取单元的数据进特征提取处理,得到中间特征数据,包括:

响应于确定与所述位置对应的数据处理单元包括特征提取单元和下采样单元,且所述下采样单元位于特征提取单元之前,由所述特征提取单元,对所述下采样单元输出的下采样特征数据进行特征提取处理,得到中间特征数据。

8.根据权利要求2-7任一项所述的方法,其中,所述基于所述子特征数据,生成所述主干网络对应的特征数据,包括:

响应于确定所述目标子网络的位置为所述至少两层子网络的最后一层子网络,基于所述目标子网络对应的子特征数据,生成所述主干网络对应的特征数据。

下载完整专利技术内容需要扣除积分,VIP会员可以免费下载。

该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京地平线信息技术有限公司,未经北京地平线信息技术有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服

本文链接:http://www.vipzhuanli.com/pat/books/202210520731.3/1.html,转载请声明来源钻瓜专利网。

同类专利
专利分类
×

专利文献下载

说明:

1、专利原文基于中国国家知识产权局专利说明书;

2、支持发明专利 、实用新型专利、外观设计专利(升级中);

3、专利数据每周两次同步更新,支持Adobe PDF格式;

4、内容包括专利技术的结构示意图流程工艺图技术构造图

5、已全新升级为极速版,下载速度显著提升!欢迎使用!

请您登陆后,进行下载,点击【登陆】 【注册】

关于我们 寻求报道 投稿须知 广告合作 版权声明 网站地图 友情链接 企业标识 联系我们

钻瓜专利网在线咨询

周一至周五 9:00-18:00

咨询在线客服咨询在线客服
tel code back_top