Aerial object tracking algorithm for UAVs based on dual-attention shuffling

被引:0
|
作者
Jin G. [1 ]
Xue Y. [1 ]
Tan L. [1 ]
Xu J. [1 ]
机构
[1] School of Nuclear Engineering, Rocket Force University of Engineering, Xi’an
基金
中国国家自然科学基金;
关键词
attention module; object tracking; region proposal network; shuffle; unmanned aerial vehicle;
D O I
10.13700/j.bh.1001-5965.2021.0177
中图分类号
学科分类号
摘要
A multi-scale real-time tracking algorithm for unmanned aerial vehicle (UAV) based on dual-attention shuffling is proposed to solve the problems of small size, large scale variation and similar object interference which often occur during UAV object tracking. First, considering the small number of target pixels in the UAV view, a deep network with double sampling integration is constructed, which provides semantic information-rich depth features and preserves the target’s detailed information. Next, a dual-attention shuffling module is designed. Channel attention and spatial attention are simultaneously grouped to filter the extracted feature information, and then the information between different channels is shuffled to enhance information exchange and improve the discriminative ability of the algorithm. Finally, to utilize the feature information of different layers, multiple region proposal networks are added to complete the target classification and regression, and the results are weighted and fused for the UAV target characteristics. Results show that the success and precision rates of the algorithm are 60.3% and 79.3% on the dataset, respectively, with 37.5 frame/s. The algorithm discrimination ability and multi-scale adaptation are significantly enhanced, which can effectively deal with the common challenges in UAV tracking. © 2023 Beijing University of Aeronautics and Astronautics (BUAA). All rights reserved.
引用
收藏
页码:53 / 65
页数:12
相关论文
共 38 条
  • [1] MENG L, YANG X., A survey of object tracking algorithms, Acta Automatica Sinica, 45, 7, pp. 1244-1260, (2019)
  • [2] BERTINETTO L, VALMADRE J, HENRIQUES J F, Et al., Fully-convolutional siamese networks for object tracking, Proceedings of the 2016 European Conference on Computer Vision (ECCV), 9914, pp. 850-865, (2016)
  • [3] HUANG C, LUCEY S, RAMANAN D., Learning policies for adaptive tracking with deep feature cascades, Proceedings of the 2017 IEEE International Conference on Computer Vision (ICCV), pp. 105-114, (2017)
  • [4] LI B, YAN J, WU W, Et al., High performance visual tracking with Siamese region proposal network, Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 8971-8980, (2018)
  • [5] REN S, HE K, GIRSHICK R, Et al., Faster R-CNN: Towards real-time object detection with region proposal networks, IEEE Transactions on Pattern Analysis and Machine Intelligence, 39, 6, pp. 1137-1149, (2017)
  • [6] LI B, WU W, WANG Q, Et al., SiamRPN++: Evolution of Siamese visual tracking with very deep networks, Proceedings of the 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 4277-4286, (2019)
  • [7] FU C, CAO Z, LI Y, Et al., Siamese anchor proposal network for high-speed aerial tracking
  • [8] LIU F, SUN Y N, WANG H J, Et al., Adaptive UAV target tracking algorithm based on residual learning, Journal of Beijing University of Aeronautics and Astronautics, 46, 10, pp. 1874-1882, (2020)
  • [9] LIU F, YANG A Z, WU Z W., Adaptive Siamese network based UAV target tracking algorithm, Acta Aeronautica et Astronautica Sinica, 41, 1, pp. 243-255, (2020)
  • [10] HE A, LUO C, TIAN X, Et al., A twofold Siamese network for real-time object tracking, Proceedings of the 2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 4834-4843, (2018)