基于人工智能网络的机器视觉目标定位方法

1.本发明属于机器视觉目标定位领域，具体涉及一种适用于复杂场景环境下的基于人工智能网络的机器视觉目标定位方法。


背景技术：

2.随着国家基础设施建设水平的快速发展，复杂场景对跟踪定位的要求连续提高，人工智能的跟踪定位技术也随之长足发展。目标定位是人工智能机器视觉的重要组成部分，该技术是指利用视频、图像的逻辑信息，对目标的轨迹和轮廓特征信息进行建模，从而对目标定位进行预测和跟踪。随着人工智能机器深度学习在图像和目标检测中的系统化使用，大量使用在目标定位方法中。但是由于图像清晰度、对比度差、背景复杂度等因素，都不利于图像的高清处理，影响了图像特征的提取精度，提升目标定位的难度。所以在此领域为科研人员提出了如何进一步减低计算成本，提升定位精度和鲁棒性的技术问题。
3.近年来，由于深度特征对目标拥有极强的表征能力，因此在人工智能的目标定位方法中有极大比例是基于深度学习的。专利文献“一种基于图像系统的车辆特征深度学习识别轨迹跟踪方法”对于滤波方法进行了改进，但是有可能带来计算量增大但不能满足实时性需求的风险。专利文献“一种目标识别方法及装置”，涉及到通过深度学习神经网络来确定目标的方法，定位速度较慢，容易发生跟踪漂移。
4.针对环境复杂，图像清晰度和对比度差，工作时对于遮挡影响较大的问题，需要提出一种适用于复杂场景环境下的基于人工智能网络的机器视觉目标定位方法。


技术实现要素：

5.本发明的目的在于提供一种解决目标定位正确率低的问题，并且降低定位成本，提高鲁棒性的基于人工智能网络的机器视觉目标定位方法。
6.本发明的目的是这样实现的，基于人工智能网络的机器视觉目标定位方法，包括如下步骤：
7.(1)检测待定位目标，采集目标图像，将采集出的目标图像为样板图像，设样板图像帧作为视频序列的第一帧；
8.(2)采用全卷积网络的孪生网络对样板图像和视频序列进行特征提取；
9.(3)通过候选网络的分类支路和回归支路对提取的特征进行分类回归；
10.(4)计算每帧视频序列的定位边界框相似度，根据定位边界框相似度进行排名，通过非极大值抑制得到每帧视频序列最后的定位边界框，如果每一帧视频序列最后的定位边界框的值都大于边界框阈值参数，则判定为定位成功，继续执行步骤(5)；如果每一帧视频序列最后的定位边界框的值中至少有一个小于等于定位边界框阈值，则判定为定位不成功，重新执行步骤(3)；
11.(5)检测视频序列的编号第g-a帧、第g-a 1帧、
…
、第g帧的视频序列中相邻两帧的定位边界框，判断定位边界框的几何中心点在xy坐标系下的平均位移变化，预测第g 1帧视
频序列相对于第g帧视频序列的移动方向，以第g帧视频序列定位边界框的几何中心点坐标为中心进行放大，形成放大到视频序列的图像尺寸的图，第g帧视频序列图像和大后的图的重叠部分，判定为第g 1帧视频序列的图像的目标定位范围。
12.所述的孪生网络分为样板支路和检测支路，两个支路的权重参数相同；其中样板支路接收样板图像，输入标记为m；检测支路接收检测序列当前帧的图像，输入标记为k；样板图像和检测序列经过孪生网络后得到孪生网络特征输出为样板图像孪生网络特征输出ι(m)和检测序列孪生网络特征输出ι(k)。
13.所述的步骤(3)具体包括如下步骤：
14.(3.1)采用fast-rcnn的损失函数对候选网络进行训练，分类支路的损失函数采用交叉熵损失函数，回归支路的损失函数采用回归损失函数smooth l1 loss；
15.(3.1.1)确认候选网络中锚箱的中心点坐标(n
x
,ny)、宽nw和高nh；
16.(3.1.2)确认实际定位边界框的中心点坐标(g
x
,gy)、高gh和宽gw；
17.(3.1.3)计算锚箱与定位边界框的标准化距离：
[0018][0019][0020][0021][0022]
(3.1.4)构建smooth l1 loss函数：
[0023][0024]
v∈[0,1,2,3]；
[0025]
ζ为边界框阈值；
[0026]
(3.1.5)回归支路的损失函数为：
[0027][0028]
(3.2)通过训练后的候选网络的两个卷积层将ι(m)扩展为样板图像孪生网络特征输出分类分支ι(m)
cls
和样板图像孪生网络特征输出回归分支ι(m)
reg
；
[0029]
(3.3)通过训练后的候选网络的两个卷积层将ι(k)扩展为检测序列孪生网络特征输出分类分支ι(k)
cls
和检测序列孪生网络特征输出回归分支ι(k)
reg
；
[0030]
(3.4)确认分类支路的通道向量个数为2x，确认回归支路的通道向量个数为4x；
[0031]
(3.5)对分类支路进行卷积运算，运算结果为：
[0032]
[0033]
(3.6)对回归支路进行卷积运算，运算结果为：
[0034][0035]
所述步骤(4)具体包括：
[0036]
(4.1)计算当前视频序列最后的定位边界框的高宽比e，以及上一帧视频序列最后的定位边界框的高宽比e"，确认当前视频序列最后的定位边界框的面积尺寸f和上一帧视频序列最后的定位边界框的面积尺寸f"；
[0037]
f2＝(gh c)(gw c)；
[0038][0039]
(4.2)计算惩罚函数：
[0040][0041]
(4.3)计算分类支路评价值：
[0042][0043]
(4.4)计算定位边界框相似度pscore：
[0044]
pscore＝penalty
×
score。
[0045]
所述的步骤(5)包括：
[0046]
以定位边界框的左上角为原点建立xy坐标轴，记录从第g-a帧开始相邻两帧的定位边界框的几何中心点坐标沿x轴方向的位移变化δxj＝[δx1,δxa]，j∈[1,a]，沿y轴方向位移变化分别为δyj＝[δy1,δya]，j∈[1,a]，计算x轴方向和y轴方向的平均位移变化：
[0047]
x轴方向平均位移变化为：
[0048][0049]
y轴方向平均位移变化为：
[0050][0051]
arg指的是复数的辐角主值；
[0052]
当x
p
》0，预测第g 1帧位置是x轴正方向；否则，预测第g 1帧位置是x轴负方向；
[0053]
当x
p
》0，预测第g 1帧位置是y轴正方向；否则，预测第g 1帧位置是y轴负方向；
[0054]
然后以第g帧对应定位边界框的几何中心点坐标为中心向外扩充为第g帧图像大小，第g帧图像和扩充后的图像的重叠部分为第g 1帧目标定位范围。
[0055]
本发明的有益效果在于：
[0056]
与现有技术相比，本发明提出了一种适用于复杂环境下的基于人工智能网络的机器视觉目标定位方法。针对目前人工智能连续定位迫切需求，提出基于相似度的评价机制，以判别继续定位的必要性，根据目标前几帧视频图像轨迹变化，来预测下一帧目标定位范围的预测策略，以有效地缩小复杂情况下目标定位的范围，减少图像精度、对比度对目标定
位造成的错误定位、丢失目标和定位漂移等影响，保证目标定位的鲁棒性。
附图说明
[0057]
图1是定位流程图；
[0058]
图2是定位示意图中第g-2帧的定位结果；
[0059]
图3是定位示意图中第g-1帧的定位结果；
[0060]
图4是定位示意图中第g帧的定位结果；
[0061]
图5是定位示意图中第g 1帧的定位结果；
[0062]
图6是连续定位图。
具体实施方式
[0063]
下面结合附图与具体实施方式对本发明作进一步详细描述。
[0064]
图1为定位流程图。首先检测待定位的目标，采集目标图像，将采集出的目标图像为样板图像，将样板图像帧作为视频序列的第一帧，作为样板支路的输入，并将所定位的视频序列逐帧作为检测支路的输入，通过孪生网络提取特征，提取后的特征图作为输入经过两个卷积层扩展成为分类分支和回归分支，进行相似度值得计算，若相似度值大于等于设定阈值参数，则认为定位成功，如果相似度值小于设定的阈值参数，则重新检测目标，且下一帧的定位范围相对于上一帧的定位范围要进行缩小，来提高定位的鲁棒性。本发明包括如下步骤：
[0065]
(1)检测待定位目标，采集目标图像，将采集出的目标图像为样板图像，设样板图像帧作为视频序列的第一帧；
[0066]
(2)采用全卷积网络的孪生网络对样板图像和视频序列进行特征提取；
[0067]
(3)通过候选网络的分类支路和回归支路对提取的特征进行分类回归；
[0068]
(4)计算每帧视频序列的定位边界框相似度，根据定位边界框相似度进行排名，通过非极大值抑制得到每帧视频序列最后的定位边界框，如果每一帧视频序列最后的定位边界框的值都大于边界框阈值参数，则判定为定位成功，继续执行步骤(5)；如果每一帧视频序列最后的定位边界框的值中至少有一个小于等于定位边界框阈值，则判定为定位不成功，重新执行步骤(3)；
[0069]
(5)检测视频序列的编号第g-a帧、第g-a 1帧、
…
、第g帧的视频序列中相邻两帧的定位边界框，判断定位边界框的几何中心点在xy坐标系下的平均位移变化，预测第g 1帧视频序列相对于第g帧视频序列的移动方向，以第g帧视频序列定位边界框的几何中心点坐标为中心进行放大，形成放大到视频序列的图像尺寸的图，第g帧视频序列图像和大后的图的重叠部分，判定为第g 1帧视频序列的图像的目标定位范围。
[0070]
1.本发明的区别特征1在于通过候选网络对特征进行分类回归，对定位边界框相似度进行排名，通过前帧图像的趋势变化预判图像的目标定位范围的发明思路，方法与现有技术相比能够效地缩小复杂情况下目标定位的范围。
[0071]
所述的孪生网络分为样板支路和检测支路，两个支路的权重参数相同；其中样板支路接收样板图像，输入标记为m；检测支路接收检测序列当前帧的图像，输入标记为k；样板图像和检测序列经过孪生网络后得到孪生网络特征输出为样板图像孪生网络特征输出ι
(m)和检测序列孪生网络特征输出ι(k)。
[0072]
所述的步骤(3)具体包括如下步骤：
[0073]
(3.1)采用fast-rcnn的损失函数对候选网络进行训练，分类支路的损失函数采用交叉熵损失函数，回归支路的损失函数采用回归损失函数smooth l1 loss；
[0074]
(3.1.1)确认候选网络中锚箱的中心点坐标(n
x
,ny)、宽nw和高nh；
[0075]
(3.1.2)确认实际定位边界框的中心点坐标(g
x
,gy)、高gh和宽gw；
[0076]
(3.1.3)计算锚箱与定位边界框的标准化距离：
[0077][0078][0079][0080][0081]
(3.1.4)构建smooth l1 loss函数：
[0082][0083]
v∈[0,1,2,3]；
[0084]
ζ为边界框阈值；
[0085]
(3.1.5)回归支路的损失函数为：
[0086][0087]
(3.2)通过训练后的候选网络的两个卷积层将ι(m)扩展为样板图像孪生网络特征输出分类分支ι(m)
cls
和样板图像孪生网络特征输出回归分支ι(m)
reg
；
[0088]
(3.3)通过训练后的候选网络的两个卷积层将ι(k)扩展为检测序列孪生网络特征输出分类分支ι(k)
cls
和检测序列孪生网络特征输出回归分支ι(k)
reg
；
[0089]
(3.4)确认分类支路的通道向量个数为2x，确认回归支路的通道向量个数为4x；
[0090]
(3.5)对分类支路进行卷积运算，运算结果为：
[0091][0092]
(3.6)对回归支路进行卷积运算，运算结果为：
[0093][0094]
所述步骤(4)具体包括：
[0095]
(4.1)计算当前视频序列最后的定位边界框的高宽比e，以及上一帧视频序列最后的定位边界框的高宽比e"，确认当前视频序列最后的定位边界框的面积尺寸f和上一帧视
频序列最后的定位边界框的面积尺寸f"；
[0096]
f2＝(gh c)(gw c)；
[0097][0098]
(4.2)计算惩罚函数：
[0099][0100]
(4.3)计算分类支路评价值：
[0101][0102]
(4.4)计算定位边界框相似度pscore：
[0103]
pscore＝penalty
×
score。
[0104]
所述的步骤(5)包括：
[0105]
以定位边界框的左上角为原点建立xy坐标轴，记录从第g-a帧开始相邻两帧的定位边界框的几何中心点坐标沿x轴方向的位移变化δxj＝[δx1,δxa]，j∈[1,a]，沿y轴方向位移变化分别为δyj＝[δy1,δya]，j∈[1,a]，计算x轴方向和y轴方向的平均位移变化：
[0106]
x轴方向平均位移变化为：
[0107][0108]
y轴方向平均位移变化为：
[0109][0110]
arg指的是复数的辐角主值
[0111]
当x
p
》0，预测第g 1帧位置是x轴正方向；否则，预测第g 1帧位置是x轴负方向；
[0112]
当x
p
》0，预测第g 1帧位置是y轴正方向；否则，预测第g 1帧位置是y轴负方向；
[0113]
然后以第g帧对应定位边界框的几何中心点坐标为中心向外扩充为第g帧图像大小，第g帧图像和扩充后的图像的重叠部分为第g 1帧目标定位范围。
[0114]
2.本发明的区别特征2在于使用损失函数对候选网络进行训练，通过孪生网络利用分类支路的损失函数和回归支路的损失函数进行输出，并利用其计算定位边界框相似度并进行目标定位范围的确认。与现有技术相比，其能够进一步提高定位精度并避免环境干扰。
[0115]
综上，本发明针对目标通过孪生网络对样板图像和定位的视频序列进行逐帧的特征提取，通过候选网络区分目标的环境变化，再对定位框回进行归定位，来提高目标定位的准确率；然后通过相似度比较制度，根据结果判断再次定位的必要性，来实现对移动目标的持续定位。本发明在环境影响较大，图像清晰度和对比度较差等干扰因素下提高目标定位的成功精度，保证了对复杂环境下目标持续长久定位。
[0116]
实施例1
[0117]
结合图1，本发明包括：
[0118]
步骤(1)，特征提取：首先采用目标识别算法检测和识别目标，例如待定位的汽车，将检测到的汽车图像作为样板图像，采用无填充全卷积网络孪生网络对样板图像和视频序列逐帧进行特征提取；孪生网络分为两个支路：样板支路和检测支路；样板支路接收样板图像，输入标记为m；检测支路接收当前帧的图像，输入标记为k。
[0119]
样板图像孪生网络特征输出ι(m)和检测序列孪生网络特征输出ι(k)为孪生网络特征图的输出，两个网络输入不同，权重参数相同。
[0120]
步骤(2)分类回归：采用候选网络的两个支路对目标定位进行分类回归。
[0121]
候选网络选取多个锚箱的锚点进行枚举并共享卷积特征，该方法使本发明获得高精度的同时还提高了时间效率。由于环境分类和定位边界框回归的监测，使候选网络更精确的提取特征。候选网络分为两个支路：分类支路和回归支路。分类支路来区分图片中的环境信息；回归支路进行坐标回归，准确设置定位框的位置和大小。对于分类支路，如果特征图生成x个锚点，在候选网络中通道数上升了2x倍，x＝4。对于回归支路，每个锚点输出边界框的四个坐标点信息，通道数上升了4x倍。通过两个卷积层将样板输出扩展成分类分支和回归分支，检测输出也被两个卷积层扩展成为分类分支和回归分支，对分类支路进行卷积运算，运算结果为：
[0122][0123]
对回归支路进行卷积运算，运算结果为：
[0124][0125]
还代表锚点对应原始图像的正激活和负激活；还代表回归支路的卷积运算结果。采用fast-rcnn的损失函数对候选网络进行训练，分类支路的损失函数采用交叉熵损失函数，回归支路的损失函数采用回归损失函数smooth l1 loss；对于回归支路的损失函数，首先将锚点的坐标标准化。计算锚箱与定位边界框的标准化距离：
[0126][0127][0128][0129][0130]
smooth l1 loss函数为：
[0131][0132]
v∈[0,1,2,3]；
[0133]
ζ为边界框阈值；
[0134]
回归支路的损失函数为：
[0135][0136]
步骤(3)相似度判断：计算当前视频序列最后的定位边界框的高宽比e，以及上一帧视频序列最后的定位边界框的高宽比e"，确认当前视频序列最后的定位边界框的面积尺寸f和上一帧视频序列最后的定位边界框的面积尺寸f"；
[0137]
f2＝(gh c)(gw c)
[0138][0139]
计算惩罚函数：
[0140][0141]
计算分类支路评价值：
[0142][0143]
计算定位边界框相似度pscore：
[0144]
pscore＝penalty
×
score。
[0145]
将当前帧的分类支路评价值乘以惩罚函数，对当前帧的生成边界框评估，使用非最大抑制得到当前帧的最终定位边界框。若pscore》0.9，则认为定位成功，无需进行重新跟踪。此机制来判断目标定位是否需要接着进行，以提高跟踪的鲁棒性。
[0146]
步骤(4)预测定位范围：在目标定位工作时，由于各种因素扰动导致图像质量下降，造成目标丢失、定位漂移、错误定位等情况的发生，为应对这些影响，根据前几帧的定位结果预测下一帧的定位范围，将目标下一帧的轨迹方向设定为与上一帧的运动方向一致，此定位策略可有效地缩小目标定位的范围。
[0147]
由于定位目标运动的速度变化，帧与帧之间时间间隔很短，因此在图像范围内目标的位置变化会存在惯性效应，本方法根据前几帧对目标定位的结果，判断下一帧目标运动方向。以定位边界框的左上角为原点建立xy坐标轴，记录从第g-a帧开始相邻两帧的定位边界框的几何中心点坐标沿x轴方向的位移变化δxj＝[δx1,δxa]，j∈[1,a]，沿y轴方向位移变化分别为δyj＝[δy1,δya]，j∈[1,a]，计算x轴方向和y轴方向的平均位移变化：
[0148]
x轴方向平均位移变化为：
[0149][0150]
y轴方向平均位移变化为：
[0151][0152]
arg指的是复数的辐角主值。
[0153]
当x
p
》0，预测第g 1帧位置是x轴正方向；否则，预测第g 1帧位置是x轴负方向；
[0154]
当x
p
》0，预测第g 1帧位置是y轴正方向；否则，预测第g 1帧位置是y轴负方向；
[0155]
然后以第g帧对应定位边界框的几何中心点坐标为中心向外扩充为第g帧图像大小，第g帧图像和扩充后的图像的重叠部分为第g 1帧目标定位范围。
[0156]
图2-5为定位效果示意图。其中图2、图3、图4和图5分别表示第g-2帧、第g-1帧、第g帧和第g 1帧的的定位结果。以视频序列中每一帧图像的左上角为原点建立坐标轴，由此判断在g 1帧时目标的位置在当前第g帧目标位置的左上方。以上一帧定位框的中心点，向外扩充为与当前帧的图像大小，两图重叠部分，即为下一帧目标定位的选择范围。图6为连续跟踪图，进一步表现了本方法在长时连续定位的效果。实验过程中，持续对凸点目标跟踪了300秒，共5700帧图片，其中每秒19帧图片。综上，本发明方法与现有技术相比区别明显，具有实质性的技术效果。