[发明专利]一种基于关键点回归的多物体检测方法、系统、终端和存储介质在审
申请号: | 201810834358.2 | 申请日: | 2018-07-26 |
公开(公告)号: | CN109190467A | 公开(公告)日: | 2019-01-11 |
发明(设计)人: | 吴子章;王凡;唐锐;李坤仑;丁丽珠 | 申请(专利权)人: | 北京纵目安驰智能科技有限公司 |
主分类号: | G06K9/00 | 分类号: | G06K9/00;G06K9/32;G06K9/46;G06K9/62 |
代理公司: | 暂无信息 | 代理人: | 暂无信息 |
地址: | 102200 北京市昌平区*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 关键点 追踪 特征图 相对位置信息 回归 存储介质 第二位置 第一位置 检测区域 损失函数 网络结构 物体检测 预设 标签 终端 目标检测区域 多尺度特征 无干扰 减小 融合 检测 优化 | ||
1.一种基于关键点回归的多物体检测方法,其特征在于,包括以下步骤:
S01:预设追踪目标的关键点标签,确定追踪目标的检测区域,获得预设追踪目标关键点标签在检测区域中的相对位置信息,标记为关键点第一位置信息;
S02:提取目标检测区域的特征图,获取追踪目标关键点在特征图中的相对位置信息,标记为关键点第二位置信息;
S03:以关键点第一位置信息和关键点第二位置信息为输入,得出损失函数以优化网络结构。
2.根据权利要求1所述的基于关键点回归的多物体检测方法,其特征在于,步骤S01中所述追踪目标的类别为至少两种类别,所述预设追踪目标的关键点标签按照其预设的追踪目标类别、以及每个类别中每个预设的追踪目标个体分别预设关键点;所述步骤S01中确定追踪目标的检测区域时,根据追踪目标的类别不同分别确定每个类别中每个追踪目标个体所在的检测区域,再将每个预设的追踪目标个体的关键点标签与追踪目标个体所在的检测区域相匹配;分别得到各个追踪目标个体所对应关键点信息。
3.根据权利要求2所述的基于关键点回归的多物体检测方法,其特征在于,所述步骤S02中,提取目标检测区域的特征图后,还包括将各个特征图的特征融合的步骤S021。
4.根据权利要求3所述的基于关键点回归的多物体检测方法,其特征在于,在步骤S01的确定追踪目标的检测区域后,在步骤S02的提取目标检测区域的特征图前,还包括尺度空间转换的步骤S01a:将各个目标检测区域的特征图转换成相同的尺度的标准特征图,再进行关键点检测,获取关键点在特征图中的相对位置关系。
5.根据权利要求4所述的基于关键点回归的多物体检测方法,其特征在于,在获取追踪目标关键点在特征图中的相对位置信息前,还包括将标准特征图下采样的步骤S01b:获取下采样层特征图,以下采样层特征图作为输入,训练下采样局部网络;再以下采样层特征图为输入下采样局部网络,输出关键点位置信息并映射回标准特征图中。
6.根据权利要求5所述的基于关键点回归的多物体检测方法,其特征在于,还包括步骤S01c:将映射了下采样关键点位置的标准特征图进行掩码操作,训练标准特征局部网络,使其只学习标准特征图中关键点所在的映射位置。
7.根据权利要求1所述的基于关键点回归的多物体检测方法,其特征在于,所述网络结构中,基础段采用resnet50网络结构,检测段采用rrc网络结构。
8.根据权利要求1所述的基于关键点回归的多物体检测方法,其特征在于,所述关键点检测段的网络结构包括获取基础段中低层的特征图,通过RoI pooling层使每个特征图的窗口生成固定尺寸的特征图,以concat函数融合固定尺寸的特征图,再经过至少一次卷积、池化操作获得标准特征图,标准特征图与预设追踪目标的关键点标签输入一起生成第一损失函数;所述标准特征图再经过至少一次卷积、池化操作获得下采样层特征图,下采样层特征图和追踪目标关键点在特征图中的标签共同作为输入再通过掩码操作,生成第二损失函数。
9.一种基于关键点回归的多物体检测系统,其特征在于,包括关键点标签标注模块、目标检测模块、特征提取模块、关键点第一位置生成模块、关键点第二位置信息生成模块、损失函数生成模块;
所述目标检测模块用于在原图中获得追踪目标,并以追踪目标为基础获得检测区域;
所述关键点标签标注模块用于标注追踪目标关键点,并输出关键点标签;
所述特征提取模块用于自检测区域中提取特征,生成特征图;
所述关键点第一位置生成模块用于以关键点标签所在目标检测模块的像素点的位置信息生成关键点第一位置数组;
所述关键点第二位置生成模块用于以关键点标签所在特征图的格点的位置信息生成关键点第二位置数组;
所述损失函数生成模块用于以关键点第一位置数组与第二位置数组对应位数之差的和与系数的乘积得到损失函数,以修正网络结构。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京纵目安驰智能科技有限公司,未经北京纵目安驰智能科技有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201810834358.2/1.html,转载请声明来源钻瓜专利网。
- 上一篇:一种人员实时定位的方法和设备
- 下一篇:一种疲劳驾驶监测方法及系统