[发明专利]一种目标跟踪定位方法、装置及计算机可读介质

权利要求书

1.一种目标跟踪定位方法，其特征在于，包括：

在获取到视频序列的视频图像时，确定所述视频图像中的模板帧和搜索帧；其中，所述模板帧为所述视频序列中的第一帧的视频图像，所述模板帧中包括待跟踪目标及其位置，所述搜索帧为所述视频序列中第一帧之后任一帧的视频图像；

将所述模板帧和所述搜索帧均输入至卷积神经网络模型的孪生网络框架中的模板分支及搜索分支中；

在所述模板分支及所述搜索分支中，对所述模板帧和所述搜索帧进行特征提取，并对特征提取后的所述模板帧和所述搜索帧进行卷积操作，得到响应图；

利用通道注意力机制处理所述响应图，得到通道注意力特征图；

将所述通道注意力特征图和所述响应图做点乘生成第一特征图；

利用空间注意力机制处理所述第一特征图，得到空间注意力特征图；

将所述空间注意力特征图和所述第一特征图做点乘生成第二特征图；

基于所述模板分支得到的所述第二特征图和所述搜索分支得到的所述第二特征图确定所述搜索帧中所述待跟踪目标及其位置；

所述对所述模板帧和所述搜索帧进行特征提取，包括：

对所述模板帧和所述搜索帧进行卷积操作;

所述孪生网络框架为SiamRPN。

2.如权利要求1所述的目标跟踪定位方法，其特征在于，确定所述视频图像中的模板帧和所述搜索帧，包括：

判断所述模板帧的大小是否小于模板尺寸；

若所述模板帧的大小小于所述模板尺寸，对所述模板帧进行颜色填充以使填充后的所述模板帧的尺寸达到所述模板尺寸；

若所述模板帧的大小不小于所述模板尺寸，以所述模板帧的中心点为基准对所述模板帧的四周进行裁剪，以使裁剪后的所述模板帧的尺寸达到所述模板尺寸；

判断所述搜索帧的大小是否小于搜索尺寸；

若所述搜索帧的大小小于所述搜索尺寸，对所述搜索帧进行颜色填充以使填充后的所述搜索帧的尺寸达到所述搜索尺寸；

若所述搜索帧的大小不小于所述搜索尺寸，以上一帧的所述视频图像中的所述待跟踪目标的中心点为基准对所述搜索帧的四周进行裁剪，以使裁剪后的所述搜索帧的尺寸达到所述搜索尺寸。

3.如权利要求2所述的目标跟踪定位方法，其特征在于，对所述模板帧进行颜色填充以使填充后的所述模板帧的尺寸达到所述模板尺寸，包括：

利用所述模板帧的众数RGB值对所述模板帧进行颜色填充；

对所述搜索帧进行颜色填充以使填充后的所述搜索帧的尺寸达到所述搜索尺寸，包括：

利用所述搜索帧的众数RGB值对所述搜索帧进行颜色填充。

4.如权利要求2所述的目标跟踪定位方法，其特征在于，对所述模板帧进行颜色填充以使填充后的所述模板帧的尺寸达到所述模板尺寸，包括：

利用所述模板帧的平均RGB值对所述模板帧进行颜色填充；

对所述搜索帧进行颜色填充以使填充后的所述搜索帧的尺寸达到所述搜索尺寸，包括：

利用所述搜索帧的平均RGB值对所述搜索帧进行颜色填充。

5.如权利要求1所述的目标跟踪定位方法，其特征在于，所述卷积神经网络模型为采用MobileNetV2构建的骨架网络。

6.如权利要求1所述的目标跟踪定位方法，其特征在于，对所述模板帧和搜索帧进行特征提取，包括：

基于所述模板帧和搜索帧的纹理和梯度进行特征提取。

7.一种目标追踪定位装置，其特征在于，包括：

存储器，用于存储计算机程序；

处理器，用于执行所述计算机程序时实现如权利要求1至6任一项所述的目标追踪定位方法的步骤。

8.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有计算机程序，所述计算机程序被处理器执行时实现如权利要求1至6任一项所述的目标追踪定位方法的步骤。