Computer Science ›› 2023, Vol. 50 ›› Issue (6A): 220500005-6.doi: 10.11896/jsjkx.220500005

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Electiric Bike Helment Wearing Detection Alogrithm Based on Improved YOLOv5

XIE Puxuan1,2, CUI Jinrong1,2, ZHAO Min3   

  1. 1 College of Mathematics and Informatics,South China Agricultural University,Guangzhou 510642,China;
    2 Guangzhou Key Laboratory of Intelligent Agriculture,Guangzhou 510642,China;
    3 Shenzhen Institute of Artificial Intelligence and Robotics,Shenzhen,Guangdong 518129,China
  • Online:2023-06-10 Published:2023-06-12
  • About author:XIE Puxuan,born in 2000,undergra-duate.His main research interests include machine learning and image processing. CUI Jinrong,born in ,Ph.D,lecture,is a member of China Computer Federation.Her main research interests include face recognition,image processing and pattern recognition,machine learning etc.
  • Supported by:
    National Key R&D Program of China(2019YFB1310402) and National Natural Science Foundation of China(1,62172285).

Abstract: In electric vehicle traffic accidents,craniocerebral injury is the main cause of death of electric vehicle riders,and most electric vehicle riders rarely wear helmets.Therefore,it is of strong practical significance to supervise the helmet wearing situation of electric vehicle riders by combining the target detection algorithm with road cameras.For the current problems of electric vehicle helmet wearing detection:the high leakage rate of targets blocking each other,and the high leakage rate of smaller targets,this paper proposes an improved YOLOv5 target detection algorithm to achieve the detection of electric vehicle helmet wearing.The method first adds the channel attention mechanism ECA-Net to the YOLOv5 network,so that the model can detect the target features,thus improving the model detection performance;the Bi-FPN weighted bidirectional feature pyramid module is used toachieve a balance of the importance of features at different levels,which is conducive to improving the small target miss detection problem;the loss function of Alpha-CIoU Loss is used to improve the accuracy of model localization.Experimental results show that the detection accuracy of the method is higher than other models for the helmet wearing situation of electric vehicle riders in all three scenarios,with an average accuracy of 95.8%,which is higher than the original network detection accuracy,and achieves high accuracy detection of electric vehicle helmet wearing situation.

Key words: Deep learning, Helmet wearing detection, YOLOv5, Bi-FPN, ECA-Net, Alpha-CIoU

CLC Number: 

  • TP391
[1]The Ministry of Public Security Traffic Mana-gement Bureau deployed to carry out the “one h-elmet and one belt” safety guarding action[J].Road Traffic Management,2020,35(5):9-10.
[2]LIU C,WANG J T,WANG M Y.Application of SSD network with introduced visual mechanis-m in motorcycle helmet wearing detection[J].Journal of Electronic Measurement and Instrumentation,2021,35(3):144-151.
[3]DING M D,LI Y X.Design and implementation of helmet wea-ring recognition system for cycl-ists based on YOLOv3[J].China New Communication,2021,23(13):44-45.
[4]JIA J S,BAO Q J,TANG H M.Safety helmet wearing detection based on deformable component model[J].Computer Application Research,2016,33(3):953-956.
[5]CHENG C C,HU S F.Research on intelligent detection of helmet wearing in intensive scenes[J].Intelligent Computers and Applications,2020,10(9):223-224.
[6]REDMON J,DIVVALA S,GIRSHICK R,et al.You Only Look Once:Unified,Real-Time Objct Detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2016:779-788.
[7]LIU W,ANGUELOV D,ERHAN D,et al.SSD:Single ShotMultiBox Detector[C]//Computer Vision-ECCV 2016:14th European Conference.Springer International Publishing,2016:21-37.
[8]GIRSHICK R.Fast R-CNN[C]//Proceedings of the IEEE International Conference on Computer Vision.2015:1440-1448.
[9]TAN M,PANG R,LE Q V.EfficientDet:S-calable and Efficient Object Detection[C]//2020 IEEE/C-VF Conference on Computer Vision and Pattern Recognition(CVPR).IEEE,2020.
[10]HE J,ERFANI S,MA X,et al.Alpha-IoU:A Family of Power Intersection over Union Losses for Bounding Box Regression[J].Advances in Neural Information Processing Systems,2021,34:20230-20242.
[11]WANG Q,WU B,ZHU P,et al.ECA-Net:Efficie-nt Channel Attention for Deep Convolutional Neu-ral Networks[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).IEEE,2020.
[12]LIN T Y,DOLLAR P,GIRSHICKR,et al.Feature P-yramidNetworks for Object Detection[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2017:2117-2125.
[13]LI H,XIONG P,AN J,et al.Pyramid Attention Network for Semantic Segmentation[J].arXiv:1805.10180,2018.
[14]JIE H,LI S,GANG S,et al.Squeeze-and-Excitation Networks[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.2018:7132-7141.
[15]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.
[1] WANG Yu, WANG Zuchao, PAN Rui. Survey of DGA Domain Name Detection Based on Character Feature [J]. Computer Science, 2023, 50(8): 251-259.
[2] ZHANG Yian, YANG Ying, REN Gang, WANG Gang. Study on Multimodal Online Reviews Helpfulness Prediction Based on Attention Mechanism [J]. Computer Science, 2023, 50(8): 37-44.
[3] SONG Xinyang, YAN Zhiyuan, SUN Muyi, DAI Linlin, LI Qi, SUN Zhenan. Review of Talking Face Generation [J]. Computer Science, 2023, 50(8): 68-78.
[4] WANG Xu, WU Yanxia, ZHANG Xue, HONG Ruize, LI Guangsheng. Survey of Rotating Object Detection Research in Computer Vision [J]. Computer Science, 2023, 50(8): 79-92.
[5] ZHOU Ziyi, XIONG Hailing. Image Captioning Optimization Strategy Based on Deep Learning [J]. Computer Science, 2023, 50(8): 99-110.
[6] ZHANG Xiao, DONG Hongbin. Lightweight Multi-view Stereo Integrating Coarse Cost Volume and Bilateral Grid [J]. Computer Science, 2023, 50(8): 125-132.
[7] LI Kun, GUO Wei, ZHANG Fan, DU Jiayu, YANG Meiyue. Adversarial Malware Generation Method Based on Genetic Algorithm [J]. Computer Science, 2023, 50(7): 325-331.
[8] WANG Mingxia, XIONG Yun. Disease Diagnosis Prediction Algorithm Based on Contrastive Learning [J]. Computer Science, 2023, 50(7): 46-52.
[9] SHEN Zhehui, WANG Kailai, KONG Xiangjie. Exploring Station Spatio-Temporal Mobility Pattern:A Short and Long-term Traffic Prediction Framework [J]. Computer Science, 2023, 50(7): 98-106.
[10] HUO Weile, JING Tao, REN Shuang. Review of 3D Object Detection for Autonomous Driving [J]. Computer Science, 2023, 50(7): 107-118.
[11] ZHOU Bo, JIANG Peifeng, DUAN Chang, LUO Yuetong. Study on Single Background Object Detection Oriented Improved-RetinaNet Model and Its Application [J]. Computer Science, 2023, 50(7): 137-142.
[12] MAO Huihui, ZHAO Xiaole, DU Shengdong, TENG Fei, LI Tianrui. Short-term Subway Passenger Flow Forecasting Based on Graphical Embedding of Temporal Knowledge [J]. Computer Science, 2023, 50(7): 213-220.
[13] LI Yuqiang, LI Linfeng, ZHU Hao, HOU Mengshu. Deep Learning-based Algorithm for Active IPv6 Address Prediction [J]. Computer Science, 2023, 50(7): 261-269.
[14] GAO Xiang, TANG Jiqiang, ZHU Junwu, LIANG Mingxuan, LI Yang. Study on Named Entity Recognition Method Based on Knowledge Graph Enhancement [J]. Computer Science, 2023, 50(6A): 220700153-6.
[15] ZENG Wu, MAO Guojun. Few-shot Learning Method Based on Multi-graph Feature Aggregation [J]. Computer Science, 2023, 50(6A): 220400029-10.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!