Outline

Ingegneria Sismica

Ingegneria Sismica

Deep Learning Based Target Detection in UAV Images-Drone-YOLO Algorithm

Author(s): Jingping Guo1
1China People’s Police University (Guangzhou), Guangzhou, Guangdong, 510663, China
Guo, Jingping. “Deep Learning Based Target Detection in UAV Images-Drone-YOLO Algorithm.” Ingegneria Sismica Volume 43 Issue 1: 1-26, doi:10.65102/is2026411.

Abstract

Along with the evolution of scientific knowledge, UAV aerial photography technology has become an important tool for acquiring information, and the research of UAV target detection will contribute to the decision-making and guidance of traffic management and diversion, with great prospects in the future field of intelligent transportation. This paper proposes an algorithm model of improved YOLOv5 (Drone-YOLO) for target detection based on the expansion of UAV image data by utilizing the improved cyclic generative adversarial network (CycleGAN). The Drone-YOLO algorithm model improves the adaptability to the size changes of UAV image data targets through the increase of detection branches, integration of multi-level information, and fusion of multi-scale features. Additionally, the introduction of the multi-scale attention mechanism increases the attention ability of the network on the target objects; the decoupling of classification and regression tasks promotes the detection accuracy, and the optimization of the loss function improves the efficiency of training. According to the experiments conducted on the drone image data set of VisDrone, the precision of the Drone-YOLO algorithm reaches 51.5%, an increase of 5.6% from the original model. Meanwhile, the recall rate was improved from 34.2% to 39.6%, mAP@.5 from 34.5% to 39.8%, and mAP@.5:.95 from 18.3% to 23.1%, all of which are enough to complete the detection task of target objects in UAV scenarios.

 

Keywords
generative adversarial networks; multi-scale attention; target detection; YOLO algorithm; UAVs

Related Articles

Zhihao Jiang1,2, Limi Chen1,2, Jing Yang1
1Hainan Vocational University of Science and Technology, Haikou 571126, China
2Institute for Mathematical Research, Universiti Putra Malaysia, Serdang 43400, Malaysia
Limi Chen1,2, Zhihao Jiang1,2, Jing Yang1
1Hainan Vocational University of Science and Technology, Haikou 571126, China
2Institute for Mathematical Research, Universiti Putra Malaysia, Serdang 43400, Malaysia
Hui Yuan1, Minjie Chai2, Siqing Xu1, Jinsong Li1, Jinwan Zheng1
1Electric Power Research Institute, State Grid Shanxi Electric Power Co., Ltd., Taiyuan, 030001, Shanxi, China
2Jincheng Power Supply Branch, State Grid Shanxi Electric Power Co., Ltd., Jincheng, 048000, Shanxi, China
Yanhan Zhu1,2
1China Academy of Cultural Heritage, Chaoyang District, 100029, Beijing, China
2Beijing University of Civil Engineering and Architecture, Xicheng District, 100044, Beijing, China
Ken Wang1, Jinhan Shu2, Kan Yuan1
1School of Digital Media, Shenzhen Polytechnic University, Shenzhen 518055, Guangdong, China
2Postdoctoral Mobile Station of Journalism and communication, Fudan University, Shanghai 200433, Shanghai, China