[发明专利]特征金字塔网络的训练方法、提取图像特征的方法和装置有效
申请号: | 201910779263.X | 申请日: | 2019-08-22 |
公开(公告)号: | CN110516732B | 公开(公告)日: | 2022-03-15 |
发明(设计)人: | 郭超旭;王国利;张骞 | 申请(专利权)人: | 北京地平线机器人技术研发有限公司 |
主分类号: | G06V10/774 | 分类号: | G06V10/774;G06V10/25;G06V10/82;G06V20/58;G06V40/10;G06K9/62 |
代理公司: | 北京思源智汇知识产权代理有限公司 11657 | 代理人: | 毛丽琴 |
地址: | 100080 北京*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 特征 金字塔 网络 训练 方法 提取 图像 装置 | ||
1.一种特征金字塔网络的训练方法,包括:
获取样本图像的兴趣区域、以及待训练的特征金字塔网络中的用于产生多尺度特征的隐含层针对所述样本图像输出的第一金字塔特征;其中,所述第一金字塔特征中的最高层特征用于与至少一上下文特征生成所述特征金字塔网络针对所述样本图像输出的第二金字塔特征中的最高层特征,所述至少一上下文特征从输入金字塔特征中的最高层特征中获取的、包含有所述兴趣区域内的特征以及所述兴趣区域外一定范围内的特征,所述输入金字塔特征为输入所述特征金字塔网络的、所述样本图像的金字塔特征;
根据所述兴趣区域,从所述第一金字塔特征中获取第一兴趣区域特征;
对所述第一兴趣区域特征进行目标对象检测,获得第一目标对象检测结果;
根据所述第一目标对象检测结果与样本图像的目标对象标注信息之间的差异,调整所述特征金字塔网络的网络参数。
2.根据权利要求1所述的方法,其中,所述对所述第一兴趣区域特征进行目标对象检测,获得第一目标对象检测结果,包括:
对所述第一兴趣区域特征进行目标对象检测,获得目标对象位置信息和目标对象类别信息;
所述根据所述第一目标对象检测结果与样本图像的目标对象标注信息之间的差异,调整所述特征金字塔网络的网络参数,包括:
根据所述目标对象位置信息与样本图像的目标对象位置标注信息之间的差异、以及所述目标对象类别信息与样本图像的目标对象类别标注信息之间的差异确定出的损失,从所述隐含层进行反向传播,调整所述特征金字塔网络的网络参数。
3.根据权利要求1或2所述的方法,其中,所述方法还包括:
获取所述特征金字塔网络针对所述样本图像输出的第二金字塔特征;
从所述第二金字塔特征的多层特征中分别获取所述兴趣区域对应的多个第二兴趣区域特征;
对所述多个第二兴趣区域特征进行特征融合,获得兴趣区域融合特征;
对所述兴趣区域融合特征进行目标对象检测,获得第二目标对象检测结果;
根据所述第二目标对象检测结果与所述样本图像的目标对象标注信息之间的差异,调整所述特征金字塔网络的网络参数。
4.根据权利要求3所述的方法,其中,所述对所述多个第二兴趣区域特征进行特征融合,获得兴趣区域融合特征,包括:
将所述多个第二兴趣区域特征转换为具有相同空间分辨率的多个第二兴趣区域特征;
根据所述具有相同空间分辨率的多个第二兴趣区域特征以及所述多个第二兴趣区域特征各自对应的第一权重,进行特征融合处理,形成兴趣区域融合特征。
5.根据权利要求4所述的方法,其中,所述特征金字塔网络输出所述样本图像的第二金字塔特征的过程,包括:
从所述特征金字塔网络的输入金字塔特征中的最高层特征中,获取至少一上下文特征;其中,所述输入金字塔特征用于形成所述第一金字塔特征;
根据所述至少一上下文特征和所述第一金字塔特征中的最高层特征,生成所述样本图像的第二金字塔特征中的最高层特征。
6.根据权利要求5所述的方法,其中,所述至少一上下文特征包括:多个空间分辨率不同的上下文特征;
所述根据所述至少一上下文特征和所述第一金字塔特征的最高层特征,生成所述样本图像的第二金字塔特征的最高层特征,包括:
将所述多个空间分辨率不同的上下文特征转换为多个具有相同空间分辨率的上下文特征;
根据所述转换后的多个上下文特征以及各自对应的第二权重,进行特征融合处理,形成融合上下文特征;
将所述第一金字塔特征中的最高层特征和所述融合上下文特征叠加,获得所述样本图像的第二金字塔特征的最高层特征。
7.根据权利要求6所述的方法,其中,所述第一权重或者第二权重包括:
针对特征中的多个空间位置分别设置的权重;或者
针对特征的不同通道分别设置的权重;或者
针对特征中的多个空间位置处的不同通道分别设置的权重。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京地平线机器人技术研发有限公司,未经北京地平线机器人技术研发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201910779263.X/1.html,转载请声明来源钻瓜专利网。