Enforcing Traffic Safety: A Deep Learning Approach for Detecting Motorcyclists' Helmet Violations Using YOLOv8 and Deep Convolutional Generative Adversarial Network-Generated Images

被引:4
作者
Shoman, Maged [1 ]
Ghoul, Tarek [1 ]
Lanzaro, Gabriel [1 ]
Alsharif, Tala [1 ]
Gargoum, Suliman [1 ]
Sayed, Tarek [1 ]
机构
[1] Univ British Columbia, Dept Civil Engn, Vancouver, BC V6T 1Z4, Canada
关键词
deep learning; DCGANs; YOLOv8; helmet detection; imbalanced classes; VULNERABLE ROAD USERS; CRASHES; INJURY; FATALITIES; RISK;
D O I
10.3390/a17050202
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In this study, we introduce an innovative methodology for the detection of helmet usage violations among motorcyclists, integrating the YOLOv8 object detection algorithm with deep convolutional generative adversarial networks (DCGANs). The objective of this research is to enhance the precision of existing helmet violation detection techniques, which are typically reliant on manual inspection and susceptible to inaccuracies. The proposed methodology involves model training on an extensive dataset comprising both authentic and synthetic images, and demonstrates high accuracy in identifying helmet violations, including scenarios with multiple riders. Data augmentation, in conjunction with synthetic images produced by DCGANs, is utilized to expand the training data volume, particularly focusing on imbalanced classes, thereby facilitating superior model generalization to real-world circumstances. The stand-alone YOLOv8 model exhibited an F1 score of 0.91 for all classes at a confidence level of 0.617, whereas the DCGANs + YOLOv8 model demonstrated an F1 score of 0.96 for all classes at a reduced confidence level of 0.334. These findings highlight the potential of DCGANs in enhancing the accuracy of helmet rule violation detection, thus fostering safer motorcycling practices.
引用
收藏
页数:16
相关论文
共 48 条
[1]   A Vision-based System for Traffic Anomaly Detection using Deep Learning and Decision Trees [J].
Aboah, Armstrong ;
Shoman, Maged ;
Mandal, Vishal ;
Davami, Sayedomidreza ;
Adu-Gyamfi, Yaw ;
Sharma, Anuj .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS, CVPRW 2021, 2021, :4202-4207
[2]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, 10.48550/arXiv.2004.10934, DOI 10.48550/ARXIV.2004.10934]
[3]   Risk factors to driver fatalities in single-vehicle crashes: Comparisons between non-motorcycle drivers and motorcyclists [J].
Chang, HL ;
Yeh, TH .
JOURNAL OF TRANSPORTATION ENGINEERING, 2006, 132 (03) :227-236
[4]   Powered two-wheeler riders' risk of crashes associated with filtering on urban roads [J].
Clabaux, Nicolas ;
Fournier, Jean-Yves ;
Michel, Jean-Emmanuel .
TRAFFIC INJURY PREVENTION, 2017, 18 (02) :182-187
[5]   An analysis of the injury severity of motorcycle crashes in Brazil using mixed ordered response models [J].
Cunto, Flavio J. C. ;
Ferreira, Sara .
JOURNAL OF TRANSPORTATION SAFETY & SECURITY, 2017, 9 :33-46
[6]  
Dahiya K, 2016, IEEE IJCNN, P3046, DOI 10.1109/IJCNN.2016.7727586
[7]   Histograms of oriented gradients for human detection [J].
Dalal, N ;
Triggs, B .
2005 IEEE COMPUTER SOCIETY CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, VOL 1, PROCEEDINGS, 2005, :886-893
[8]  
Desai M, 2016, INT J ENG TRENDS TEC, V35, P185, DOI [10.14445/22315381/IJETT-V35P241, DOI 10.14445/22315381/IJETT-V35P241]
[9]   Half and Full Helmet Wearing Detection in Thailand using Haar Like Feature and Circle Hough Transform on Image Processing [J].
Doungmala, Pathasu ;
Klubsuwan, Katanyoo .
2016 IEEE INTERNATIONAL CONFERENCE ON COMPUTER AND INFORMATION TECHNOLOGY (CIT), 2016, :611-614
[10]   INJURY PATTERNS IN MOTORCYCLE COLLISIONS [J].
DRYSDALE, WF ;
KRAUS, JF ;
FRANTI, CE ;
RIGGINS, RS .
JOURNAL OF TRAUMA-INJURY INFECTION AND CRITICAL CARE, 1975, 15 (02) :99-115