Computer Science ›› 2025, Vol. 52 ›› Issue (6A): 240900167-8.doi: 10.11896/jsjkx.240900167

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Improved Helmet Detection Algorithm of Electric Bicycle Based on YOLOv8n

BAI Shaokang, WANG Baohui, CHEN Jixuan   

  1. Beihang University,Beijing 100080,China
  • Online:2025-06-16 Published:2025-06-12
  • About author:BAI Shaokang,born in 1991,postgra-duate.His main research interests include computer graphics and nerual networks.
    WANG Baohui,born in 1973,senior engineer,master supervisor.His main research interests include software architecture,big data,artificial intelligence,etc.

Abstract: With the continuous advancement of the transportation industry,the necessity of helmet-wearing during the riding of electric bicycles has been consistently verified.Meanwhile,the detection of helmet-wearing by electric bicycle riders has also received extensive research in the domain of deep learning.Currently,there exist issues such as significant detection difficulties in dense scenarios and challenges in detecting small targets when it comes to helmet-wearing by electric bicycle riders.Concurrently,in order to achieve more effective deployment on mobile terminals,a lightweight model needs to be selected.Hence,an improved detection algorithm for helmet-wearing by electric bicycle riders based on YOLOv8n has been proposed.Firstly,YOLOv8n is the most lightweight model among the YOLOv8 series,enabling better deployment on mobile devices.Additionally,a switchable atrous convolution is introduced into the backbone network of YOLOv8n,enhancing the feature extraction capability of YOLOv8n in dense scenes without increasing the computational load.At the end of the image pyramid feature fusion network of YOLOv8n,a triple attention mechanism is integrated to strengthen the extraction ability of important features from the fused information of different-scale features by the YOLOv8n model.Finally,the fusion of large-sized feature information with small-sized feature information is added to improve the detection effect of small targets.Ultimately,while ensuring the lightweight nature of YOLOv8n,when using the improved model in the self-constructed validation set,the mAP@50,mAP@50-90,and recall rate have increased by 2.7%,2.8% and 2.5%,respectively.Therefore,the proposed method holds certain practical and scientific significance.

Key words: Electric bicycle, Helemet detection, YOLOv8n, Empty convolution, Attention mechanis

CLC Number: 

  • TP312
[1]YU Y,PENG J J,LI Q F,et al.Helmet wearing among e-bikers in Shanghai:On-site observational study[J].ShanghaiJournal of Preventive Medicine,2018(9):730-735
[2]LU X,CHEN D,LIU W C,et al.Characteristics of road traffic injuries caused by electric bicycles in Suzhou during 2017-2019[J].J Trauma Surg,2021(5):375-379.
[3]KRIZHEVSKY A,SUTSKEVER I,HINTON G.ImageNetClassification with Deep Convolutional Neural Networks[J].Advances in Neural Information Processing Systems,2012,25(2).
[4]GIRSHICK R,DONAHUE J,DARRELL T,et al,Rich feature hierarchies for accurate object detection and semantic segmentation[C]//Proceedings of the 2014 IEEE Conferenceon ComputerVision and Pattern Recognition.Columbus:IEEE,2014:580-587.
[5]REN S,HE K,GIRSHICK R,et al.Faster R-CNN:Towards Real-Time Object Detection with Region Proposal Networks[J].IEEE Transactions on Pattern Analysis & Machine Intelligence,2017,39(6):1137-1149.
[6]LIN T Y,DOLLAR P,GIRSHICK R,et al.Feature Pyramid Networks for Object Detection[C]//CVPR.2017:936-944.
[7]HE K,GKIOXARI G,PIOTR D,et al.Mask R-CNN[J].IEEE Transactions on Pattern Analysis & Machine Intelligence,2017.2980-2988.
[8]GE S M,GUO W J,LI C Y,et al.Masked Face Recognition with Generative-to-Discriminative Representations[J].arXiv:2405.16761,2024.
[9]WANG G,YUAN Y,LI J,et al.Receptive Multi-GranularityRepresentation for Person Re-Identification[C]//IEEE Transactions on Image Processing.2020:6096-6109.
[10]JIJI C V,SHINE L.Automated detection of helmet on motorcyclists from traffic surveillance videos:a comparative analysis using hand-crafted features and CNN[J].Multimedia Tools and Applications,2020:1-21.
[11]MISRAA A K,MANOHAR NMISTRY J,AGARWAL M,et al.An automatic detection of helmeted and non-helmeted motorcyclist with license plate extraction using convolutional neural network[C]//Seventh International Conference on Image Processing Theory,Tools and Applications(IPTA).IEEE,2017:1-6.
[12]CHATTERJI S,DASGUPTA M,BANDYOPADHYAY O.Automated helmet detection for multiple motorcycle riders using CNN[C]//IEEE Conference on Information and CommunicationTechnology.IEEE,2019:1-4.
[13]XU H,DENG Z H.Helmet Wearing Detection Algorithm byImproved YOLOv5[J],Software Guide,2024,8:33-42.
[14]ZHU X,LYU S,WANG X,et al.TPH-YOLOv5:ImprovedYOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-captured Scenarios[J].arXiv.2108.11539,2021.
[15]XU C G.Rearch real-time detection algorithm for helmets and reflective vests based on YOLOv5s[J].Computer Era,2023,10:110-115.
[16]HAN K,WANG Y,TIAN Q,et al.GhostNet:More Features From Cheap Operations[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).IEEE,2020.
[17]WANG J,CHEN K,XU R,et al.CARAFE:Content-Aware ReAssembly of Features[C]//2019 IEEE/CVF International Conference on Computer Vision.2019.
[18]QIAO S,CHEN L C,YUILLE A.DetectoRS:Detecting Objects with Recursive Feature Pyramid and Switchable Atrous Convolution[J].arXiv:2006.02334,2020.
[19]MISRA D,NALAMADA T,ARASANIPALAI A U,et al.Rotate to Attend:Convolutional Triplet Attention Module[J].ar-Xiv:2010.03045,2020.
[20]HU J,SHEN L,SUN G,et al.Squeeze-and-Excitation Networks[C]//2018 IEEE/CVF Conference on Computer Vision and Patern Recognition.2018.
[21]WOO S,PARK J,LEE J Y,et al.CBAM:Convolutional Block Attention Module[J].arXiv:1807.06521,2018.
[22]PARK J,WOO S,LEE J Y,et al.BAM:Bottleneck Attention Module[J].arXiv:1807.06514,2018.
[1] LI Daicheng, LI Han, LIU Zheyu, GONG Shiheng. MacBERT Based Chinese Named Entity Recognition Fusion with Dependent Syntactic Information and Multi-view Lexical Information [J]. Computer Science, 2025, 52(6A): 240600121-8.
[2] HUANG Bocheng, WANG Xiaolong, AN Guocheng, ZHANG Tao. Transmission Line Fault Identification Method Based on Transfer Learning and Improved YOLOv8s [J]. Computer Science, 2025, 52(6A): 240800044-8.
[3] WU Zhihua, CHENG Jianghua, LIU Tong, CAI Yahui, CHENG Bang, PAN Lehao. Human Target Detection Algorithm for Low-quality Laser Through-window Imaging [J]. Computer Science, 2025, 52(6A): 240600069-6.
[4] ZENG Fanyun, LIAN Hechun, FENG Shanshan, WANG Qingmei. Material SEM Image Retrieval Method Based on Multi-scale Features and Enhanced HybridAttention Mechanism [J]. Computer Science, 2025, 52(6A): 240800014-7.
[5] HOU Zhexiao, LI Bicheng, CAI Bingyan, XU Yifei. High Quality Image Generation Method Based on Improved Diffusion Model [J]. Computer Science, 2025, 52(6A): 240500094-9.
[6] DING Xuxing, ZHOU Xueding, QIAN Qiang, REN Yueyue, FENG Youhong. High-precision and Real-time Detection Algorithm for Photovoltaic Glass Edge Defects Based onFeature Reuse and Cheap Operation [J]. Computer Science, 2025, 52(6A): 240400146-10.
[7] ZHAO Chanchan, YANG Xingchen, SHI Bao, LYU Fei, LIU Libin. Optimization Strategy of Task Offloading Based on Meta Reinforcement Learning [J]. Computer Science, 2025, 52(6A): 240800050-8.
[8] WANG Rong , ZOU Shuping, HAO Pengfei, GUO Jiawei, SHU Peng. Sand Dust Image Enhancement Method Based on Multi-cascaded Attention Interaction [J]. Computer Science, 2025, 52(6A): 240800048-7.
[9] WANG Baohui, GAO Zhan, XU Lin, TAN Yingjie. Research and Implementation of Mine Gas Concentration Prediction Algorithm Based on Deep Learning [J]. Computer Science, 2025, 52(6A): 240400188-7.
[10] ZHENG Chuangrui, DENG Xiuqin, CHEN Lei. Traffic Prediction Model Based on Decoupled Adaptive Dynamic Graph Convolution [J]. Computer Science, 2025, 52(6A): 240400149-8.
[11] HONG Yi, SHEN Shikai, SHE Yumei, YANG Bin, DAI Fei, WANG Jianxiao, ZHANG Liyi. Multivariate Time Series Prediction Based on Dynamic Graph Learning and Attention Mechanism [J]. Computer Science, 2025, 52(6A): 240700047-8.
[12] TENG Minjun, SUN Tengzhong, LI Yanchen, CHEN Yuan, SONG Mofei. Internet Application User Profiling Analysis Based on Selection State Space Graph Neural Network [J]. Computer Science, 2025, 52(6A): 240900060-8.
[13] GUAN Xin, YANG Xueyong, YANG Xiaolin, MENG Xiangfu. Tumor Mutation Prediction Model of Lung Adenocarcinoma Based on Pathological [J]. Computer Science, 2025, 52(6A): 240700010-8.
[14] TAN Jiahui, WEN Chenyan, HUANG Wei, HU Kai. CT Image Segmentation of Intracranial Hemorrhage Based on ESC-TransUNet Network [J]. Computer Science, 2025, 52(6A): 240700030-9.
[15] CHEN Xianglong, LI Haijun. LST-ARBunet:An Improved Deep Learning Algorithm for Nodule Segmentation in Lung CT Images [J]. Computer Science, 2025, 52(6A): 240600020-10.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!