Helmet Wearing Detection Algorithm Intergrating Transfer Learning and YOLOv5

Cao Zhi-xiong; Wu Xiao-ling; Luo Xiao-wei; Ling Jie

doi:10.12052/gdutxb.220139

English Version Journal Login

您当前的位置：

首页 >

文章列表页 >

Helmet Wearing Detection Algorithm Intergrating Transfer Learning and YOLOv5

更新时间：2026-05-27

- Helmet Wearing Detection Algorithm Intergrating Transfer Learning and YOLOv5
- Vol. 40, Issue 4, Pages: 67-76(2023)
- 作者机构：
  
  1. 广东工业大学计算机学院,广东,广州,510006
  2. 香港城市大学建筑及土木系,香港,999077
- 作者简介：
- 基金信息：
- DOI：10.12052/gdutxb.220139
  CLC： TP391.41
- Online First：25 July 2023，
  
  Published：2023
- 稿件说明：
移动端阅览
曹智雄, 吴晓鸰, 骆晓伟, 凌捷. 融合迁移学习与YOLOv5的安全帽佩戴检测算法[J]. 广东工业大学学报, 2023, 40(4): 67-76.

Cao Zhi-xiong, Wu Xiao-ling, Luo Xiao-wei, et al. Helmet Wearing Detection Algorithm Intergrating Transfer Learning and YOLOv5[J]. 2023, 40(4): 67-76.
曹智雄, 吴晓鸰, 骆晓伟, 凌捷. 融合迁移学习与YOLOv5的安全帽佩戴检测算法[J]. 广东工业大学学报, 2023, 40(4): 67-76. DOI： 10.12052/gdutxb.220139.

Cao Zhi-xiong, Wu Xiao-ling, Luo Xiao-wei, et al. Helmet Wearing Detection Algorithm Intergrating Transfer Learning and YOLOv5[J]. 2023, 40(4): 67-76. DOI： 10.12052/gdutxb.220139.

摘要

针对现有安全帽佩戴检测算法在检测小目标和密集目标时出现漏检、检测准确度低下等问题，本文提出一种基于改进YOLOv5和迁移学习的安全帽佩戴检测新方法。使用K-means算法聚类出更适合检测任务的先验框尺寸以解决默认先验框不适应任务的问题；在特征提取网络后段引入空间通道混合注意力模块，使模型加强对目标权重的学习，抑制无关背景的权重；改进YOLOv5后处理阶段的非极大值抑制(Non-Maximum-Suppression，NMS) 算法的判断度量，减少预测框误删和缺失的现象；采用迁移学习的策略对网络进行训练，克服现有数据集不足的缺陷并提升模型泛化能力；最后提出一种适用于视觉传感网络的安全帽佩戴级联判断框架。实验结果表明改进模型的平均准确率(IOU=0.5)达到了93.6%，与原始模型相比提高了5%，性能优于其他同类算法，提高了施工场景下对安全帽佩戴检测的准确率。

Abstract

To address the problems of missing detection and low detection accuracy of the existing helmet wearing detection algorithms for small and crowded targets detection

this paper proposes a helmet wearing detection method based on improved YOLOv5 and transfer learning. First

different from the default priori frame that is not suitable for the task

we use the K-means algorithm to cluster the suitable priori frame size for the detection task. Then

in the back of the feature extraction network

we introduce a spatial channel mixed attention module to strengthen the learning of relevant weights and suppress the weights of irrelevant backgrounds

respectively. Further

we improve the judgment metric of the non-maximum-suppression (NMS) algorithm in the post-processing stage of YOLOv5 to reduce the phenomenon of false deletion and missing of prediction boxes. After that

the proposed network is trained based on the strategy of transfer learning

which can overcome the scarcity of limited existing data sets and improve the generalization ability of the model. Finally

we build a cascade judgment framework for helmet wearing deployed in visual sensor networks. The experimental results show that our proposed method improves the average accuracy (IOU=0.5) to 93.6%

which is 5% higher than the original model in the helmet wearing data set. The proposed model also outperforms other state-of-the-art algorithms by obviously improving the accuracy of helmet wearing detection in the construction scenarios.

关键词

Keywords

references

GUO S H, JING J R, ZHANG X D, et al . Researchon detection of safety helmet wearing basedon improved YOLOv4 [J ] . China Safety Production Science and Technology, 2021, 17(12): 135-141.

YUE S Q, ZHANG Q, SHAO D Q, et al . Safety helmat wearing status detection study based on ResNet50-SSD [J ] . Yangtze River Information and Communication, 2021, 34(3): 86-89.

GIRSHICK R, DONAHUE J, DARRELL T , et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]// CVPR 2014: Proceedings of the 2014 IEEE Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2014: 580-587.

REN S, HE K, GIRSHICK R, et al . Faster R-CNN: towards real-time object detection with region proposa l networks [J ] . IEEE Transactions on Pattern Analysis Machine Intelligence, 2017, 39(6): 1137-1149.

REDMON J, DIVVALA S, GiIRSHICK R, et al. You only look once: unified, real-time object detection[C]//Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. Piscataway: IEEE, 2016: 779-788.

REDMON J, FARHADI A. YOLO9000: better, faster, stronger[C]// IEEE Conference on Computer Vision Pattern Recognition. Piscataway: IEEE, 2017: 6517-6525.

LIN J, DANG W C, PAN L H, et al . Safety helmet detection based on YOLO [J ] . Computer System Applications, 2019, 28(9): 174-179.

FANG M, SUN T T, SHAO Z. Fast helmet-wearing-condition detection based onimproved YOLOv2 [J]. Optical Precision Engineering, 2019, 27(5): 1196-1205.

CHENG R, HE X W, ZHENG Z L, et al . Multi-scale safety helmet detection based on SAS-YOLOv3-Tiny [J ] . Applied Sciences, 2021, 11(8): 3652.

CHEN W, LIU M, ZHOU X, et al . Safety helmet wearing detection in aerial images using improved YOLOv4 [J ] . Computers, Materials, Continua, 2022(8): 16.

HUANG Z Q, LI J. Research on traffic sign recognition based on spatial channel attention mechanism and multi-scale fusion [J]. Journal of Nanjing University of Posts and Telecommunications (Natural Science Edition), 2022, 42(2): 93-102.

WOO S , PARK J , LEE J Y , et al. CBAM: convolutional block attention module[C]//Proceedings of the European Conference on Computer Vision(ECCV) . Munich: Springer, 2018: 3-19.

YING X, WANG Y, WANG L, et al . A stereo attention module for stereo image super-resolution [J ] . IEEE Signal Processing Letters, 2020, 27(99): 496-500.

ZHANG D, WU W, CHENG H, et al . Image-to-video person re-identification with temporally memorized similarity learning [J ] . IEEE Transactions on Circuits Systems for Video Technology, 2018, 28(10): 2622-2632.

Views

下载量

CSCD

Alert me when the article has been cited

提交

Tools

Publicity Resources

Efficient Temporal Modeling for RGB-T Tracking

Ground Deformation Monitoring and Influencing Factors Analysis in Fuzhou City Based on PSI Technology

The Cross-twin Physics-informed Neural Network for Wave Equations with Conserved Quantities

Bimodal Iterative Cross-attention Fusion Ensemble Framework

A Study of Power Quality Regulation Strategies of Energy Storage Converters

Related Author

No data

Related Institution

No data

AI问答

⁰