Computer Science ›› 2022, Vol. 49 ›› Issue (6A): 424-428.doi: 10.11896/jsjkx.220100252

• Image Processing & Multimedia Technology • Previous Articles     Next Articles

Real-time Helmet Detection Algorithm Based on Circumcircle Radius Difference Loss

CHEN Yong-ping1, ZHU Jian-qing1,2, XIE Yi1, WU Han-xiao3, ZENG Huan-qiang1   

  1. 1 College of Engineering,Huaqiao University,Quanzhou,Fujian 362021,China
    2 College of Information Science and Engineering,Huaqiao University,Xiamen,Fujian 361021,China
    3 Xiamen Yealink Network Technology Co.,LTD,Xiamen,Fujian 361015,China
  • Online:2022-06-10 Published:2022-06-08
  • About author:CHEN Yong-ping,born in 1998,postgraduate.His main research interests include image processing and deep learning.
    ZHU Jian-qing,born in 1987,professor.His main research interests include deep learning,machine vision and pattern recognition.
  • Supported by:
    National Natural Science Foundation of China(61976098,61871434),National Key R & D Program of China(2021YFE0205400),Natural Science Foundation for Outstanding Young Scholars of Fujian Province(2019J06017) and Collaborative Innovation Platform Project of Fuxiaquan National Independent Innovation Demonstration Zone(2021FX03).

Abstract: For large demands of fast and accurate helmet detection,this paper proposes a real-time helmet detection algorithm.Firstly,to solve the gradient vanish problem of using bounding box regression loss functions,this paper proposes the circumcircle radius difference (CRD) loss function.Secondly,to solve the problem of complex multi-scale feature fusion layers restricting detection speeds,this paper proposes a lightweight focus on small object (FSO) feature fusion layer.Finally,this paper combines the YOLO network,CRD,and FSO to form a YOLO-CRD-FSO (YCF) model for real-time helmet detection.On a Jetson Xavier NX device,experiments show that the detection speed of YCF reaches 43.4 frames per second for 640×640 sized videos,which is nearly 2 frames per second faster than the state-of-the-art YOLO-V5 model,and the mean average precision has been improved by nearly 1%.The proposed YCF detection model comprehensively optimizes boundary box regression loss functions and feature fusions,acquiring good helmet detection results.

Key words: Bounding box regression, Feature fusion layer, Object detection, Safety helmet detection, YOLO

CLC Number: 

  • TP391
[1] YU J,JIANG Y,WANG Z,et al.Unitbox:An Advanced Object Detection Network[C]//ACM International Conference on Multimedia.New York:ACM Press,2016:516-520.
[2] REZATOFIGHI R,TSOI N,SAVARESE S,et al.GeneralizedIntersection Over Union:A Metric and A Loss for Bounding Box Regression[C]//IEEE Conference on Computer Vision and Pattern Recognition.Piscataway:IEEE Press,2019:658-666.
[3] ZHENG Z,WANG P,LIU W,et al.Distance-IoU Loss:Faster and Better Learning for Bounding Box Regression[EB/OL].[2021-07-10].
[4] ZHENG T,ZHAO S,LIU Y,et al.SCALoss:Side and Corner Aligned Loss for Bounding Box Regression[EB/OL].[2021-07-10].
[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 and Machine Intelligence,2017,39(6):1137-1149.
[6] QIN Z,LI Z,ZHANG Z,et al.Thunder Net:Towards Real-time Generic Object Detection[EB/OL].[2021-07-10].
[7] CAI Z,VASCONCELOS N.Cascade R-CNN:Delvinginto High Quality Object Detection[C]//IEEE Conference on Computer Vision and Pattern Recognition.Piscataway:IEEE Press,2018:6154-6162.
[8] LIU Y D,WANG Y T,WANG S W,et al.CBNet:A Novel Composite Backbone Network Architecture for Object Detection[EB/OL].[2021-07-10].
[9] JOCHER G.YOLO-V5[EB/OL].[2021-01-10].
[10] LIN T Y,DOLLAR P,GIRSHICK R,et al.Feature Pyramid Networks for Object Detection[C]//IEEE Conference on Computer Vision and Pattern Recognition.Piscataway:IEEE Press,2017:936-944.
[11] LIU S,QI L,QIN H F,et al.Path Aggregation Network for Instance Segmentation[C]//IEEE Conference on Computer Vision and Pattern Recognition.Piscataway:IEEE Press,2018:8759-8768.
[12] REDMON J,DIVVALA S,GIRSHICK R,et al.You Only Look Once:Unified,Real-Time Object Detection[C]//IEEE Confe-rence on Computer Vision and Pattern Recognition.Piscataway:IEEE Press,2016:779-788.
[13] WANG C,LIAO H M,WU Y,et al.CSPNet:A New Backbone that Can Enhance Learning Capability of CNN[C]//IEEE Conference on Computer Vision and Pattern Recognition Workshops.Piscataway:IEEE Press,2020:1571-1580.
[14] LIN T Y,MAIRE M,BELONGIE S,et al.Microsoft COCO:Common Objects in Context[C]//European Conference on Computer Vision.Heidelberg:Springer,2014:740-755.
[15] ZHAO Y,YANG J,LIU M,et al.Intelligent Edge ComputingTechnology Based on Federal Learning for Video Monitoring[J].Journal on Communications,2020,41(10):109-115.
[16] REDMON J,FARHADI A.YOLOv3:An Incremental Improvement[EB/OL].[2021-07-10].
[17] HE K,ZHANG X,REN S,et al.Spatial Pyramid Pooling inDeep Convolutional Networks for Visual Recognition[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2014,37(9):1904-1916.
[1] LIU Dong-mei, XU Yang, WU Ze-bin, LIU Qian, SONG Bin, WEI Zhi-hui. Incremental Object Detection Method Based on Border Distance Measurement [J]. Computer Science, 2022, 49(8): 136-142.
[2] WANG Can, LIU Yong-jian, XIE Qing, MA Yan-chun. Anchor Free Object Detection Algorithm Based on Soft Label and Sample Weight Optimization [J]. Computer Science, 2022, 49(8): 157-164.
[3] CHU Yu-chun, GONG Hang, Wang Xue-fang, LIU Pei-shun. Study on Knowledge Distillation of Target Detection Algorithm Based on YOLOv4 [J]. Computer Science, 2022, 49(6A): 337-344.
[4] HAO Qiang, LI Jie, ZHANG Man, WANG Lu. Spatial Non-cooperative Target Components Recognition Algorithm Based on Improved YOLOv3 [J]. Computer Science, 2022, 49(6A): 358-362.
[5] CHEN Jia-zhou, ZHAO Yi-bo, XU Yang-hui, MA Ji, JIN Ling-feng, QIN Xu-jia. Small Object Detection in 3D Urban Scenes [J]. Computer Science, 2022, 49(6): 238-244.
[6] HU Fu-yuan, WAN Xin-jun, SHEN Ming-fei, XU Jiang-lang, YAO Rui, TAO Zhong-ben. Survey Progress on Image Instance Segmentation Methods of Deep Convolutional Neural Network [J]. Computer Science, 2022, 49(5): 10-24.
[7] XU Tao, CHEN Yi-ren, LYU Zong-lei. Study on Reflective Vest Detection for Apron Workers Based on Improved YOLOv3 Algorithm [J]. Computer Science, 2022, 49(4): 239-246.
[8] LI Guo-quan, YAO Kai, PANG Yu. Automatic Identification Algorithm of Blood Cell Image Based on Convolutional Neural Network [J]. Computer Science, 2022, 49(4): 247-253.
[9] YUAN Lei, LIU Zi-yan, ZHU Ming-cheng, MA Shan-shan, CHEN Lin-zhou-ting. Improved YOLOv3 Remote Sensing Target Detection Based on Improved Dense Connection and Distributional Ranking Loss [J]. Computer Science, 2021, 48(9): 168-173.
[10] GONG Hao-tian, ZHANG Meng. Lightweight Anchor-free Object Detection Algorithm Based on Keypoint Detection [J]. Computer Science, 2021, 48(8): 106-110.
[11] LI Lin, LIU Xue-liang, ZHAO Ye, JI Ping. Low Light Image Fusion Detection Method Based on Lego Filter and SSD [J]. Computer Science, 2021, 48(7): 213-218.
[12] XIN Yuan-xue, SHI Peng-fei, XUE Rui-yang. Moving Object Detection Based on Region Extraction and Improved LBP Features [J]. Computer Science, 2021, 48(7): 233-237.
[13] ZHANG Man, LI Jie, ZHU Xin-zhong, SHEN Ji, CHENG Hao-tian. Augmentation Technology of Remote Sensing Dataset Based on Improved DCGAN Algorithm [J]. Computer Science, 2021, 48(6A): 80-84.
[14] YU Han-qing, YANG Zhen, YIN Zhi-jian. Tiny YOLOv3 Target Detection Algorithm Based on Region Activation Strategy [J]. Computer Science, 2021, 48(6A): 118-121.
[15] PAN Ming-yuan, SONG Hui-hui, ZHANG Kai-hua, LIU Qing-shan. Learning Global Guided Progressive Feature Aggregation Lightweight Network for Salient Object Detection [J]. Computer Science, 2021, 48(6): 103-109.
Full text



No Suggested Reading articles found!