Research on vehicle detection algorithm based on YOLOv5s-ShuffleNetV2-SE-EIOU

Huizhi Xu; Yuanming Zhang; Yinan Chen

doi:10.1117/12.3054703

20 December 2024 Research on vehicle detection algorithm based on YOLOv5s-ShuffleNetV2-SE-EIOU

Huizhi Xu, Yuanming Zhang, Yinan Chen

Author Affiliations +

Proceedings Volume 13421, Eighth International Conference on Traffic Engineering and Transportation System (ICTETS 2024); 134211A (2024) https://doi.org/10.1117/12.3054703
Event: Eighth International Conference on Traffic Engineering and Transportation System (ICTETS 2024), 2024, Dalian, China

Abstract

Vehicle detection in drone aerial imagery presents challenges such as complex background environments, small target sizes, and computational complexity, which can result in missed or false detections. To address these issues, an improved YOLOv5s-ShuffleNetV2-SE-EIOU algorithm is proposed. The backbone network is optimized using the lightweight ShuffleNetV2 module, which reduces network redundancy while increasing detection speed. The SE attention module is introduced into the backbone to enhance the model's ability to capture vehicle features, thereby improving detection accuracy in complex environments. The loss function is replaced with EIOU_loss to accelerate the convergence of prediction boxes and improve regression accuracy. Experimental results show that in vehicle detection ablation experiments, this algorithm improved mAP@0.5 detection accuracy by 2.9%, mAP@0.5:0.95 detection accuracy by 3.8%, reduced training time by 0.015 hours, and decreased parameter count by 23.4%. In special weather detection experiments, the algorithm demonstrated higher accuracy.

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

Citation Download Citation

Huizhi Xu, Yuanming Zhang, and Yinan Chen "Research on vehicle detection algorithm based on YOLOv5s-ShuffleNetV2-SE-EIOU", Proc. SPIE 13421, Eighth International Conference on Traffic Engineering and Transportation System (ICTETS 2024), 134211A (20 December 2024); https://doi.org/10.1117/12.3054703

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

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Object detection

Detection and tracking algorithms

Education and training

Ablation

Environmental sensing

Rain

Algorithm development

Show All Keywords

Keywords/Phrases

Search In:

Publication Years