融合迁移学习与YOLOv5的安全帽佩戴检测算法

曹智雄; 吴晓鸰; 骆晓伟; 凌捷

doi:10.12052/gdutxb.220139

欢迎来到能源电力期刊网！

English Version 期刊登录

您当前的位置：

首页 >

文章列表页 >

融合迁移学习与YOLOv5的安全帽佩戴检测算法

计算机科学与技术 | 更新时间：2026-05-27

- 融合迁移学习与YOLOv5的安全帽佩戴检测算法
- Helmet Wearing Detection Algorithm Intergrating Transfer Learning and YOLOv5
- 广东工业大学学报 2023年40卷第4期页码：67-76
- 作者机构：
  
  1. 广东工业大学计算机学院,广东,广州,510006
  2. 香港城市大学建筑及土木系,香港,999077
- 作者简介：
  
  曹智雄(1997–) ，男，硕士研究生，主要研究方向为智慧城市、图像处理与模式识别
  吴晓鸰(1979–) ，女，副教授，博士，主要研究方向为物联网、网络安全，E-mail：xl.wu@gdut.edu.cn
- 基金信息：
  
  教育部重点实验室开放课题(2021-1EQBD-02);广东省国际科技合作领域项目(2019A050513010)
- DOI：10.12052/gdutxb.220139
  中图分类号： TP391.41
- 网络首发：2023-07-25，
  
  纸质出版：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.

浏览量

下载量

CSCD

文章被引用时，请邮件提醒。

提交

工具集

关联资源

基于路况参数的混合动力机车能量管理策略研究

融合沟道倾斜角的3D NAND存储单元建模及编程策略优化

储能变流器电能质量治理策略研究

基于数字混频的脉冲式激光测距方法的研究与实现

基于高效时序建模的RGB-T跟踪