Aerial image detection of drone based on improved YOLOv5

Zhijun Gao; Zhi Yang; Zhonghua Han

doi:10.1117/12.2660017

3 February 2023 Aerial image detection of drone based on improved YOLOv5

Zhijun Gao, Zhi Yang, Zhonghua Han

Proceedings Volume 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022); 125111J (2023) https://doi.org/10.1117/12.2660017
Event: Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 2022, Hulun Buir, China

Abstract

Drones have been used in many practical applications,,such as agriculture, aerial photography and surveillance. Therefore, it is very difficult for machines to automatically understand the visual data collected by drones, which makes the connection between computer vision and drones become inseparable. However, the identification of targets from aerial images is faced with the difficulty that the target to be detected is usually too small, too dense, and the relative size is not fixed relative to the image, and the relative motion of drone and traffic leads to the blurred target. To address these challenges, this paper proposes an improved YOLOv5 network model. Based on YOLOv5, we integrate the convolution block attention model ( CBAM ) and the attention mechanism of Swin Transformer to enable it to effectively focus on the attention area in dense small object scene and reduce the calculation amount. We also adopt the BiFPN structure for the structure of the neck network, which can obtain more effective feature information and reduce some unnecessary connections. In addition, we also add an additional detector to detect small objects. To further improve our YOLOv5, we adopt the data enhancement strategy. Extensive experiments on the VisDrone 2019 dataset show that the The AP result of the improved YOLOv5 was 32.83 %. Compared with the baseline model ( YOLOv5 ), YOLOv5 increased by about 6.5 %, indicating that the improved model was effective.

Citation Download Citation

Zhijun Gao, Zhi Yang, and Zhonghua Han "Aerial image detection of drone based on improved YOLOv5", Proc. SPIE 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 125111J (3 February 2023); https://doi.org/10.1117/12.2660017

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
7 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Head

Target detection

Transformers

Data modeling

Neck

Sensors

Convolution

Show All Keywords

Keywords/Phrases

Search In:

Publication Years