A-YOLO: small target vehicle detection based on improved YOLOv5

Xiang Li; Shuoping Wang; Bei Wang

doi:10.1117/12.3023908

9 April 2024 A-YOLO: small target vehicle detection based on improved YOLOv5

Xiang Li, Shuoping Wang, Bei Wang

Proceedings Volume 12989, Third International Conference on Intelligent Traffic Systems and Smart City (ITSSC 2023); 129890V (2024) https://doi.org/10.1117/12.3023908
Event: Third International Conference on Intelligent Traffic Systems and Smart City (ITSSC 2023), 2023, Xi'an, China

Abstract

In this paper, we introduce an enhanced YOLOv5-based model tailored to address challenges in detecting small vehicle targets within high-altitude surveillance contexts. Within the backbone network, the SE attention mechanism is incorporated after each convolutional module. To amplify the detection capabilities, a high-resolution detection head is integrated. Within the intermediate (neck) network, we apply context aggregation and feature fusion techniques before the small target detection head, while employing the CBAM attention mechanism for other detection heads. To further optimize performance, the loss function has been revised by substituting the traditional Ciou calculation with Focal Eiou, in tandem with enhancing object localization using the OTA algorithm. Additionally, the Retinex algorithm is deployed for data augmentation. Performance assessments on the VisDrone2019 dataset revealed that our proposed approach outperformed the standard YOLOv5, 2019 challenge sota, and TPH-YOLO (with prior +CBAM) models by margins of 20%, 7%, and 3%, respectively. A series of tests affirmed the robustness of our algorithm, particularly in the realm of small vehicle target detection.

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

Citation Download Citation

Xiang Li, Shuoping Wang, and Bei Wang "A-YOLO: small target vehicle detection based on improved YOLOv5", Proc. SPIE 12989, Third International Conference on Intelligent Traffic Systems and Smart City (ITSSC 2023), 129890V (9 April 2024); https://doi.org/10.1117/12.3023908

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

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object detection

Data modeling

Detection and tracking algorithms

Head

Performance modeling

Target detection

Small targets

Show All Keywords

Keywords/Phrases

Search In:

Publication Years