UAV target detection algorithm based on improved YOLOv5s

Tao Zhang; Fenmei Wang; Dongxu Chen; Xihui Fan

doi:10.1117/12.2684583

19 October 2023 UAV target detection algorithm based on improved YOLOv5s

Tao Zhang, Fenmei Wang, Dongxu Chen, Xihui Fan

Proceedings Volume 12709, Fourth International Conference on Artificial Intelligence and Electromechanical Automation (AIEA 2023); 1270938 (2023) https://doi.org/10.1117/12.2684583
Event: Fourth International Conference on Artificial Intelligence and Electromechanical Automation (AIEA 2023), 2023, Nanjing, China

Abstract

Real-time UAV monitoring is an important means of battlefield reconnaissance, and machine interpretation of UAV images has become the main form of image interpretation, so the merit of the algorithm becomes an important factor limiting UAV reconnaissance. To address the problems of insufficient graphics card arithmetic power, low detection accuracy and difficult deployment of algorithm models at the embedded end of UAVs, this paper proposes an improved lightweight target detection algorithm based on YOLOv5s, adding K-means++ algorithm and CA attention mechanism module to the original algorithm, and training the improved YOLOv5s-CA network using tank dataset, and the simulation results show that: the improved YOLOv5s-CA has an mAP value of 97.50%, an F1 value of 0.96, and an FPS value of 74.8, which can be deployed on UAVs for real-time detection.

Citation Download Citation

Tao Zhang, Fenmei Wang, Dongxu Chen, and Xihui Fan "UAV target detection algorithm based on improved YOLOv5s", Proc. SPIE 12709, Fourth International Conference on Artificial Intelligence and Electromechanical Automation (AIEA 2023), 1270938 (19 October 2023); https://doi.org/10.1117/12.2684583

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available