中国民航大学学报 ›› 2022, Vol. 40 ›› Issue (5): 15-22.

• 民用航空 • 上一篇    下一篇

基于动态权重的双分支孪生网络目标跟踪算法

韩萍a,王皓韡b,方澄a   

  1. (中国民航大学a.电子信息与自动化学院;b.计算机科学与技术学院,天津300300)
  • 收稿日期:2021-04-12 修回日期:2021-05-18 出版日期:2022-10-15 发布日期:2023-10-27
  • 作者简介:韩萍(1966—),女,天津人,教授,博士,研究方向为信号与信息处理、SAR 目标检测与识别等.
  • 基金资助:
    中国民航大学科研启动基金项目(2017QD05S);中央高校基本科研业务费专项(3122018C005)

Object tracking algorithm based on dynamic weight in dual branch siamese network

HAN Pinga , WANG Haoweib , FANG Chenga   

  1. (a. College of Electronic Information and Automation; b. College of Computer Science and Technology, CAUC, Tianjin 300300, China)
  • Received:2021-04-12 Revised:2021-05-18 Online:2022-10-15 Published:2023-10-27

摘要: 以基于全卷积孪生网络的目标跟踪(SiamFC,fully-convolutionalsiamesenetworksforobjecttracking)算法为代表的部分深度孪生网络目标跟踪算法均是针对目标外观信息进行设计的,易受高速移动、运动模糊、光照变化等因素的影响,造成跟踪目标漂移或丢失。为了提高算法对目标外观变化的适应能力,给出一种基于动态权重的双分支孪生网络目标跟踪算法,以替换特征提取网络后的SiamFC算法作为外观分支,在此基础上增加利用双重注意力强化信息提取的语义分支作为外观分支的有效补充。跟踪阶段利用动态权重系数结合两分支的跟踪结果,有效抑制了目标外观变化对跟踪算法的影响,提升了算法的跟踪精度和鲁棒性。在4个标准目标跟踪数据集OTB2015、UAV20L、UAV123和GOT-10k上验证了本文算法的有效性,平均跟踪帧率为47帧/s,满足跟踪实时性要求。

关键词: 视频目标跟踪, 孪生网络, 注意力机制, 动态权重

Abstract: Some of siamese tracking algorithms represented by SiamFC are designed for target appearance information.However, appearance information is easily affected by factors such as high-speed movement, motion blur, and illumination changes, resulting in tracking drift or target loss. In order to improve the ability of the algorithm to adapt target appearance changes, this paper adopts a dual -branch siamese network tracking algorithm based on dynamic weights. Based on improved SiamFC algorithm as the appearance branch, a semantic branch using dual attention mechanism to enhance information extraction is added as an effective supplement. In the tracking stage, a dynamic weight is used to fuse the tracking results of the two branches, which effectively suppresses the influence of target appearance changes, also improves the tracking accuracy and robustness of the algorithm. Our algorithm has been validated on four standard object tracking data sets (OTB2015, UAV123, UAV20L, GOT-10k). The average tracking frame rate is 47 frames/s, which meets the real-time tracking requirements.

Key words: video object tracking, siamese network, attention mechanism, dynamic weight

中图分类号: