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 条
  • [11] Kwok D, 2021, NEURO-ONCOLOGY, V23, P99
  • [12] Li F., 2022, INT J MACH LEARN CYB, ppr, DOI [10.1007/s13042-021-01496-1, DOI 10.1007/S13042-021-01496-1]
  • [13] Li X., 2021, P 2021 6 INT C MULTI, P15, DOI [10.26914/c.cnkihy.2021.029103, DOI 10.26914/C.CNKIHY.2021.029103]
  • [14] Augmented Memory for Correlation Filters in Real-Time UAV Tracking
    Li, Yiming
    Fu, Changhong
    Ding, Fangqiang
    Huang, Ziyuan
    Pan, Jia
    [J]. 2020 IEEE/RSJ INTERNATIONAL CONFERENCE ON INTELLIGENT ROBOTS AND SYSTEMS (IROS), 2020, : 1559 - 1566
  • [15] Real-Time Small Drones Detection Based on Pruned YOLOv4
    Liu, Hansen
    Fan, Kuangang
    Ouyang, Qinghua
    Li, Na
    [J]. SENSORS, 2021, 21 (10)
  • [16] Security Risk Analysis in IoT Systems through Factor Identification over IoT Devices
    Omar Andrade, Roberto
    Guun Yoo, Sang
    Ortiz-Garces, Ivan
    Barriga, Jhonattan
    [J]. APPLIED SCIENCES-BASEL, 2022, 12 (06):
  • [17] Qiu X., 2021, PEDESTRIAN DETECTION
  • [18] A three-step classification framework to handle complex data distribution for radar UAV detection
    Ren, Jianfeng
    Jiang, Xudong
    [J]. PATTERN RECOGNITION, 2021, 111
  • [19] Rueda M.G.V., 2001, P AEROSPACEDEFENSE S, V4390
  • [20] Soft Computing, 2018, COMPUT WKLY NEWS