Safety helmet wearing detection algorithm based on improved YOLOv5 - CNKI
Knowledge Network Node
Journal of Computer Applications. 2022, (04) Core

Safety helmet wearing detection algorithm based on improved YOLOv5Chinese Full TextEnglish Full Text (MT)

ZHANG Jin;QU Peiqi;SUN Cheng;LUO Meng;College of Information Science and Engineering, Hunan Normal University;School of Mathematics and Statistics, Hunan Normal University;

Abstract: Aiming at the problems of strong interference and low detection precision of the existing safety helmet wearing detection,an algorithm of safety helmet detection based on improved YOLOv5（You Only Look Once version 5）model was proposed. Firstly,for the problem of different sizes of safety helmets,the K-Means++ algorithm was used to redesign the size of the anchor box and match it to the corresponding feature layer. Secondly,the multi-spectral channel attention module was embedded in the feature extraction network to ensure that the network was able to learn the weight of each channel autonomously and enhance the information dissemination between the features,thereby strengthening the network ability to distinguish foreground and background. Finally,images of different sizes were input randomly during the training iteration process to enhance the generalization ability of the algorithm. Experimental results show as follows:on the self-built safety helmet wearing detection dataset,the proposed algorithm has the mean Average Precision（mAP）reached 96. 0%,the the Average Precision（AP）of workers wearing safety helmet reached 96. 7%,and AP of workers without safety helmet reached 95. 2%. Compared with the YOLOv5 algorithm,the proposed algorithm has the mAP of helmet safety-wearing detection increased by 3. 4 percentage points,and it meets the accuracy requirement of helmet safety-wearing detection in construction scenarios.
Keywords：

safety helmet wearing detection; object detection; deep learning; YOLOv5（You Only Look Once version 5）; attention mechanism;

  • Series：

    (I) Electronic Technology ＆ Information Science; (B) Chemistry/ Metallurgy/ Environment/ Mine Industry; (C) Architecture/ Energy/ Traffic/ Electromechanics, etc

  • Subject：

    Security Science and Disaster Prevention; Architecture and Engineering; Computer Software and Application of Computer

  • Classification Code：

    TU714;TP391.41

Download the mobile appuse the app to scan this coderead the article.

Tips: Please download CAJViewer to view CAJ format full text.

Download: 5424 Page: 1292-1300 Pagecount: 9 Size: 3730K

Related Literature
  • Similar Article
  • Reader Recommendation
  • Associated Author