一种基于多图注意力机制的目标跟踪算法

摘要

本发明公开一种基于多图注意力机制的目标跟踪算法，属于一般的图像数据处理或产生技术领域，用于对视频中的目标进行跟踪，包括将视频中的第一帧图片和后续帧分别作为模板分支和搜索分支的输入，通过孪生网络对其进行特征提取；将上一步得到的输出特征输入到图注意力模块进行互相关操作；将上一步得到的输出输入到无锚的跟踪头网络中，通过分类分支得到特征图中每个像素点的分类分数，通过中心度分支得到各像素点距离目标中心的远近关系，通过回归分支得到每个像素点对应的目标框信息；将分类分数与中心度分支相乘，得到精确分类分数，找到分数最高的像素点及其对应的目标框信息，得到当前帧目标的所在位置，重复以上步骤。