A Real-Time Tracking Algorithm for Multi-Target UAV Based on Deep Learning

被引:14
作者
Hong, Tao [1 ,2 ]
Liang, Hongming [2 ]
Yang, Qiye [3 ]
Fang, Linquan [1 ]
Kadoch, Michel [4 ]
Cheriet, Mohamed [4 ]
机构
[1] Yunnan Innovat Inst BUAA, Kunming 650233, Peoples R China
[2] Beihang Univ, Sch Elect & Informat Engn, Beijing 100191, Peoples R China
[3] AVIC Chengdu Aircraft Design & Res Inst, Chengdu 610041, Peoples R China
[4] Univ Quebec, Ecole Technol Super ETS, Montreal, PQ H2L 2C4, Canada
基金
中国国家自然科学基金;
关键词
UAV; 5G; multi-target detection and tracking; YOLOv4; DeepSORT;
D O I
10.3390/rs15010002
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
UAV technology is a basic technology aiming to help realize smart living and the construction of smart cities. Its vigorous development in recent years has also increased the presence of unmanned aerial vehicles (UAVs) in people's lives, and it has been increasingly used in logistics, transportation, photography and other fields. However, the rise in the number of drones has also put pressure on city regulation. Using traditional methods to monitor small objects flying slowly at low altitudes would be costly and ineffective. This study proposed a real-time UAV tracking scheme that uses the 5G network to transmit UAV monitoring images to the cloud and adopted a machine learning algorithm to detect and track multiple targets. Aiming at the difficulties in UAV detection and tracking, we optimized the network structure of the target detector yolo4 (You Only Look Once V4) and improved the target tracker DeepSORT, adopting the detection-tracking mode. In order to verify the reliability of the algorithm, we built a data set containing 3200 pictures of four UAVs in different environments, conducted training and testing on the model, and achieved 94.35% tracking accuracy and 69FPS detection speed under the GPU environment. The model was then deployed on ZCU104 to prove the feasibility of the scheme.
引用
收藏
页数:18
相关论文
共 29 条
  • [1] Aglyamutdinova D.B., 2018, P ISPRS INT ARCH PHO, V2
  • [2] Weed detection in sesame fields using a YOLO model with an enhanced attention mechanism and feature fusion
    Chen, Jiqing
    Wang, Huabin
    Zhang, Hongdu
    Luo, Tian
    Wei, Depeng
    Long, Teng
    Wang, Zhikui
    [J]. COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2022, 202
  • [3] An Automatic Defect Detection System for Petrochemical Pipeline Based on Cycle-GAN and YOLO v5
    Chen, Kun
    Li, Hongtao
    Li, Chunshu
    Zhao, Xinyue
    Wu, Shujie
    Duan, Yuxiao
    Wang, Jinshen
    [J]. SENSORS, 2022, 22 (20)
  • [4] Chohan U.W., 2023, ACTIVIST RETAIL INVE
  • [5] DENSO TEN Limited, 2020, COMPUT NETW COMMUN
  • [6] A ResNet deep learning based facial recognition design for future multimedia applications
    Durga, B. Kanaka
    Rajesh, V.
    [J]. COMPUTERS & ELECTRICAL ENGINEERING, 2022, 104
  • [7] Fei C., 2021, REMOTE SENSING AIRCR
  • [8] A Robust Faster R-CNN Model with Feature Enhancement for Rust Detection of Transmission Line Fitting
    Guo, Zhimin
    Tian, Yangyang
    Mao, Wandeng
    [J]. SENSORS, 2022, 22 (20)
  • [9] He Yan, 2021, Journal of Physics: Conference Series, DOI 10.1088/1742-6596/2002/1/012070
  • [10] Jinhui Liu, 2021, Journal of Physics: Conference Series, V1883, DOI 10.1088/1742-6596/1883/1/012075