An Aerial Image Detection Algorithm Based on Improved YOLOv5

• Published in: Sensors (Basel, Switzerland) Vol. 24; no. 8; p. 2619
• Main Authors: Shan, Dan, Yang, Zhi, Wang, Xiaofeng, Meng, Xiangdong, Zhang, Guangwei
• Format: Journal Article
• Language: English
• Published: Switzerland MDPI AG 19-04-2024; MDPI
• Subjects: Accuracy; aerial images; Algorithms; BiFPN; Classification; Deep learning; Localization; Methods; mixed attention module; Semantics; Telecommunication systems; Unmanned aerial vehicles; YOLOv5; mixed attention module; aerial images; YOLOv5; BiFPN
• ISSN: 1424-8220; 1424-8220
• DOI: 10.3390/s24082619

To enhance aerial image detection in complex environments characterized by multiple small targets and mutual occlusion, we propose an aerial target detection algorithm based on an improved version of YOLOv5 in this paper. Firstly, we employ an improved Mosaic algorithm to address redundant boundaries arising from varying image scales and to augment the training sample size, thereby enhancing detection accuracy. Secondly, we integrate the constructed hybrid attention module into the backbone network to enhance the model's capability in extracting pertinent feature information. Subsequently, we incorporate feature fusion layer 7 and P2 fusion into the neck network, leading to a notable enhancement in the model's capability to detect small targets. Finally, we replace the original PAN + FPN network structure with the optimized BiFPN (Bidirectional Feature Pyramid Network) to enable the model to preserve deeper semantic information, thereby enhancing detection capabilities for dense objects. Experimental results indicate a substantial improvement in both the detection accuracy and speed of the enhanced algorithm compared to its original version. It is noteworthy that the enhanced algorithm exhibits a markedly improved detection performance for aerial images, particularly under real-time conditions.