• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (02): 276-282.

Previous Articles     Next Articles

An object tracking algorithm based on mixed attention mechanism

FENG Qi-yao1,2,ZHANG Jing-lei1,2   

  1. (1.School of Electrical Engineering and Automation,Tianjin University of Technology,Tianjin 300384;

    2.Tianjin Key Laboratory for Control Theory and Applications in Complicated Systems,Tianjin 300384,China)


  • Received:2020-09-21 Revised:2020-12-17 Accepted:2022-02-25 Online:2022-02-25 Published:2022-02-17

Abstract: To solve the tracking failure problem of fully-convolutional siamese networks algorithm (SiamFC) in complex scenes such as objects deformation, occlusion, and fast motion, a novel method (SiamMA) that uses the mixed attention mechanism to enhance the network identification ability is proposed. Firstly, in order to simulate the complex scenes and enhances the generalization performance of networks, an image stacking and cropping method is adopted in the network training stage to build the self-adversarial training sample pairs. Secondly, a mixed attention mechanism algorithm is proposed, which fuses spatial attention and channel attention modules in different branches of the network, so the background interference in the feature map can effectively be suppressed and the robustness of the algorithm is improved. 4 open test datasets such as Got-10k and UAV123, etc., are adopted to evaluate the algorithm performance. The experimental results show that our method outperforms 6 traditional algorithms such as SiamFC, KCF, etc., on the main performance indexes such as tracking success rate and precision. The average speed of the algorithm can reach 60 frames per second.

Key words: object tracking, siamese network, mixed attention mechanism, self-adversarial training sample pairs