基于互注意力指导的孪生跟踪算法

作者：赵越; 余志斌; 李永春

来源：计算机科学, 2021, 1-9.

• 目标跟踪
• 孪生网络
• 相似物干扰
• 互注意力模块
• 无锚框回归

针对传统孪生网络目标跟踪算法在相似物干扰、目标形变、复杂背景等跟踪环境下无法进行鲁棒跟踪的问题，提出注意力机制指导的孪生网络目标跟踪方法，以解决传统孪生跟踪方法存在的性能缺陷。首先，利用卷积神经网络ResNet50的不同网络层来提取多分辨率的目标特征，并设计互注意力模块使得模板分支与搜索分支之间的信息能够相互流动。在分类与回归网络中，将主干网络提取的每块特征信息权重参数通过神经网络自动学习、更新并加权融合每块特征的分类与回归信息。最后，根据响应图的峰值位置计算目标的预估位置和尺度信息。在UAV123数据集上，该算法相比于主流跟踪算法SiamBAN，准确率提升了1.7个点，成功率提升了0.7个点；在VOT2018数据集中，相比于SiamRPN++算法，所提算法在EAO指标上提高了2.5个点，实时跟踪速度保持在35FPS。

• 单位
  电气学院; 西南交通大学