改进YOLOv5s算法的安全帽佩戴检测研究  

Research on HelmetWearing Detection of Improved YOLOv5s Algorithm

作　　者：祁泽政 徐银霞[1,2] QI Zezheng;XU Yinxia(School of Computer Science and Engineering,Wuhan Institute of Technology,Wuhan 430205,China;Hubei Key Laboratory of Intelligent Robot,Wuhan Institute of Technology,Wuhan 430205,China)

机构地区：[1]武汉工程大学计算机科学与工程学院,武汉430205 [2]武汉工程大学智能机器人湖北省重点实验室,武汉430205

出　　处：《计算机工程与应用》2023年第14期176-183,共8页Computer Engineering and Applications

基　　金：国家自然科学基金(62171327);中国高校产学研创新基金-新一代信息技术创新项目(2020ITA07015);武汉工程大学研究生教育创新基金(CX2022356)。

摘　　要：针对安全帽佩戴检测在背景复杂、密集小目标等场景下的漏检、误检问题,基于YOLOv5s提出了一种适用于安全帽佩戴检测的改进算法。采用串联形式的混合池化优化空间金字塔池化(spatial pyramid pooling,SPP)模块,增强特征提取与特征表达,降低误检率;增加浅层输出加强深层特征图中目标位置信息的表达,提高小目标安全帽佩戴检测的性能;在切片模块中嵌入坐标注意力机制(coordinate attention,CA)扩大感受野,加强位置信息与安全帽特征的关联程度,提高安全帽目标检测的准确率。实验结果表明,改进算法的mAP达到了93.50%,比YOLOv5s、YOLOX算法分别提高了4.28、5.14个百分点,达到了复杂背景下检测小目标和密集目标的要求。Aiming at the problem of missed detection and false detection in scenarios such as complex background and dense small targets,an improved algorithm suitable for safety helmet wearing detection is proposed based on YOLOv5s.Firstly,the hybrid pooling in tandem form is used to optimize the spatial pyramid pooling(SPP)module to enhance feature extraction and feature expression and reduce the false detection rate.Then,the shallow output is added to strengthen the expression of target position information in the deep feature map,and improve the performance of small target safety helmet wearing detection.Finally,the coordinate attention mechanism(CA)is embedded in the slice module to expand the receptive field,strengthen the correlation between position information and safety helmet features,and improve the accuracy of safety helmet target detection.Experimental results show that the mAP of the improved algorithm reaches 93.50%,which is 4.28 and 5.14 percentage points higher than that of YOLOv5s and YOLOX algorithms,respectively,which meets the requirements of detecting small targets and dense targets in complex backgrounds.

关 键 词：YOLOv5s 深度学习 目标检测 人工智能 安全帽识别 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术]