Improved YOLOv5 algorithm for small object detection in UAV aerial imagery

Linqing Cai; Yanhua Yang

doi:10.1117/12.3049691

8 November 2024 Improved YOLOv5 algorithm for small object detection in UAV aerial imagery

Linqing Cai, Yanhua Yang

Author Affiliations +

Proceedings Volume 13416, Fourth International Conference on Advanced Algorithms and Neural Networks (AANN 2024); 134160G (2024) https://doi.org/10.1117/12.3049691
Event: 2024 4th International Conference on Advanced Algorithms and Neural Networks, 2024, Qingdao, China

Abstract

Due to complex backgrounds, varying object scales, and many small objects in Unmanned Aerial Vehicle (UAV) imagery, existing algorithms often have low detection accuracy, especially for small objects. To tackle these issues, we propose an improved YOLOv5-based algorithm. A Backbone Feature Weighted Fusion (BFWF) module is introduced to extract and fuse multi-scale features from the backbone. These extracted features are fused with the original neck output through cross-neck connections, enhancing fine-grained features. The neck structure is adjusted to improve the diversity and weighting of shallow fine-grained features. An Adaptive Spatial Feature Fusion (ASFF) module dynamically fuses features at different levels in the Path Aggregation Network (PANet), improving small object detection accuracy. By replacing the original loss function with the Efficient Intersection over Union (EIoU) loss function, we achieve more precise learning of small object dimensions across multi-scale scenarios. Experiments on the VisDrone2021 dataset show that the improved algorithm increases mAP50 by 3.7%, mAP75 by 2.5%, and mAP50:95 by 2.4% compared to the original YOLOv5 algorithm. This significantly enhances detection performance, making it more suitable for UAV imagery.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Linqing Cai and Yanhua Yang "Improved YOLOv5 algorithm for small object detection in UAV aerial imagery", Proc. SPIE 13416, Fourth International Conference on Advanced Algorithms and Neural Networks (AANN 2024), 134160G (8 November 2024); https://doi.org/10.1117/12.3049691

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

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
10 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object detection

Unmanned aerial vehicles

Detection and tracking algorithms

Feature fusion

Feature extraction

Airborne remote sensing

Keywords/Phrases

Search In:

Publication Years