CONFERENCE PROCEEDINGS
Advanced Search
Home > Proceedings > Volume 12923 > Article
Paper
8 November 2023 Improved electric bike helmet wearing detection algorithm for YOLOv5s
Lei Han, Hongxia Niu
Author Affiliations +
Proceedings Volume 12923, Third International Conference on Artificial Intelligence, Virtual Reality, and Visualization (AIVRV 2023); 129231Q (2023) https://doi.org/10.1117/12.3011276
Event: 3rd International Conference on Artificial Intelligence, Virtual Reality and Visualization (AIVRV 2023), 2023, Chongqing, China
Abstract
Aiming at the problems of low detection accuracy and poor robustness of electric vehicle helmets in complex traffic scenarios, an improved YOLOv5-based electric vehicle helmet detection model is proposed. First, a shallow feature detection layer is added to enable the model to better detect small targets. Then, the attention mechanism is introduced in the backbone network, which optimizes the feature extraction ability of the network. Finally, replacing the CIoU loss function with WIoU improves the convergence speed and regression accuracy. The improved YOLOv5s algorithm was trained and tested on a homemade electric vehicle helmet dataset, and the improved YOLOv5s algorithm improved the average accuracy mean, accuracy rate, and recall rate by 6%, 4%, and 6.7%, respectively, over the original algorithm. The experimental results show that the electric vehicle helmet detection model with improved YOLOv5s algorithm has high helmet detection accuracy, excellent detection of small targets, and strong generalization capability.
(2023) Published by SPIE. Downloading of the abstract is permitted for personal use only.
Citation Download Citation
Lei Han and Hongxia Niu "Improved electric bike helmet wearing detection algorithm for YOLOv5s", Proc. SPIE 12923, Third International Conference on Artificial Intelligence, Virtual Reality, and Visualization (AIVRV 2023), 129231Q (8 November 2023); https://doi.org/10.1117/12.3011276
ACCESS THE FULL ARTICLE
ORGANIZATIONAL
Sign in with credentials provided by your organization.
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
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Detection and tracking algorithms
Object detection
Small targets
Target detection
Image processing
RELATED CONTENT
Subscribe to Digital Library
Receive Erratum Email Alert
Back to Top