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

doi:10.3778/j.issn.1002-8331.2303-0284

摘要/Abstract

摘要： 针对安全帽佩戴检测在背景复杂、密集小目标等场景下的漏检、误检问题，基于YOLOv5s提出了一种适用于安全帽佩戴检测的改进算法。采用串联形式的混合池化优化空间金字塔池化（spatial pyramid pooling，SPP）模块，增强特征提取与特征表达，降低误检率；增加浅层输出加强深层特征图中目标位置信息的表达，提高小目标安全帽佩戴检测的性能；在切片模块中嵌入坐标注意力机制（coordinate attention，CA）扩大感受野，加强位置信息与安全帽特征的关联程度，提高安全帽目标检测的准确率。实验结果表明，改进算法的mAP达到了93.50%，比YOLOv5s、YOLOX算法分别提高了4.28、5.14个百分点，达到了复杂背景下检测小目标和密集目标的要求。

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

Abstract: 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.

Key words: YOLOv5s, deep learning, target detection, artificial intelligence, helmet recognition

祁泽政, 徐银霞. 改进YOLOv5s算法的安全帽佩戴检测研究[J]. 计算机工程与应用, 2023, 59(14): 176-183.

QI Zezheng, XU Yinxia. Research on Helmet Wearing Detection of Improved YOLOv5s Algorithm[J]. Computer Engineering and Applications, 2023, 59(14): 176-183.

参考文献

[1] 李政谦，刘晖.基于深度学习的安全帽佩戴检测算法综述[J].计算机应用与软件，2022，39（6）：194-202.
LI Z Q，LIU H.Helmet wearing detection algorithm based on deep learning[J].Computer Applications and Software，2022，39（6）：194-202.
[2] DAI J F，LI Y，HE K M，et al.R-FCN：object detection via region-based fully convolutional networks[C]//Proceedings of the 30th International Conference on Neural Information Processing Systems，Barcelona，Spain，Dec 5-12，2016.New York：Curran Associates Inc，2016：379-387.
[3] LU M Q，HU Y C，LU X B.Dilated light-head R-CNN using tri-center loss for driving behavior recognition[J].Image and Vision Computing，2019，90：108-121.
[4] LI J，LIANG X，SHEN S，et al.Scale-aware Fast R-CNN for pedestrian detection[J].IEEE Transactions on Multimedia，2017，20（4）：985-996.
[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 & Machine Intelligence，2017，39（6）：1137-1149.
[6] HE K，GKIOXARI G，DOLLAR P，et al.Mask R-CNN[J].IEEE Transactions on Pattern Analysis & Machine Intelligence，2020，42（2）：386-397.
[7] LIU W，ANGUELOV D，ERHAN D，et al.SSD：single shot multibox detector[C]//European Conference on Computer Vision.Cham：Springer，2016：21-37.
[8] GE Z，LIU S，WANG F，et al.YOLOX：exceeding YOLO series in 2021[J].arXiv：2107.08430，2021.
[9] DUAN K，BAI S，XIE L，et al.CenterNet：keypoint triplets for object detection[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision，2019：6569-6578.
[10] BOCHKOVSKIY A，WANG C Y，LIAO H Y M.YOLOv4：optimal speed and accuracy of object detection[C]//IEEE Conference on Computer Vision and Pattern Recognition（CVPR），2020.
[11] LIN T Y，GOYAL P，GIRSHICK R，et al.Focal loss for dense object detection[J].IEEE Transactions on Pattern Analysis & Machine Intelligence，2020，42（2）：318-327.
[12] SHRESTHA K，SHRESTHA P P，BAJRACHARYA D，et al.Hard-hat detection for construction safety visualization[J].Journal of Construction Engineering，2015，2015（1）：1-8.
[13] WU H，ZHAO J.An intelligent vision-based approach for helmet identification for work safety[J].Computers in Industry，2018，100（7）：267-277.
[14] FU D S，GAO L，HU T，et al.Research on safety helmet detection algorithm of power workers based on improved YOLOv5[J].Journal of Physics（Conference Series），2022，2171：012006.
[15] 王玲敏，段军，辛立伟.引入注意力机制的YOLOv5安全帽佩戴检测方法[J].计算机工程与应用，2022，58（9）：303-312.
WANG L M，DUAN J，XIN L W.YOLOv5 helmet wear detection method with introduction of attention mechanism[J].Computer Engineering and Applications，2022，58（9）：303-312.
[16] 吕宗喆，徐慧，杨骁，等.面向小目标的YOLOv5安全帽检测算法[J/OL].计算机应用：1-9[2023-04-05].http：//kns.cnki.net/kcms/detail/51.1307.TP.20220929.1425.005.html.
LYU Z Z，XU H，YANG X，et al.Small object detection algorithm of YOLOv5 for safety helmet[J/OL].Journal of Computer Applications：1-9[2023-04-05].http：//kns.cnki.net/kcms/detail/51.1307.TP.20220929.1425.005.html.
[17] 何丽，张红艳，房婉琳.融合多尺度边界特征的显著实例分割[J].计算机科学与探索，2022，16（8）：1865-1876.
HE L，ZHANG H Y，FANG W L.Salient instance segmentation via multicale boundary characteristic network[J].Journal of Frontiers of Computer Science and Techno-
logy，2022，16（8）：1865-1876.
[18] 朱炳宇，刘朕，张景祥.融合Grad-CAM和卷积神经网络的COVID-19检测算法[J].计算机科学与探索，2022，16（9）：2108-2120.
ZHU B Y，LIU Z，ZHANG J X.COVID-19 detection algorithm combinig Grad-CAM and convolutional neural network[J].Journal of Frontiers of Computer Science and Technology，2022，16（9）：2108-2120.
[19] HE K，ZHANG X，REN S，et al.Spatial pyramid pooling in deep convolutional networks for visual recognition[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2015，37（9）：1904-1916.
[20] LIU S，QI L，Qin H，et al.Path aggregation network for instance segmentation[C]//Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition，2018：8759-8768.
[21] ZHENG Z，WANG P，LIU W，et al.Distance-IoU loss：faster and better learning for bounding box regression[C]//Proceedings of AAAI Conference on Artificial Intelligence，2020.
[22] CHANG J，ZHANG L，GU N，et al.A mix-pooling CNN architecture with FCRF for brain tumor segmentation[J].Journal of Visual Communication & Image Representation，2019，58：316-322.
[23] WANG K，LIEW J H，ZOU Y，et al.PANet：few-shot image semantic segmentation with prototype alignment[C]//2019 IEEE/CVF International Conference on Computer Vision（ICCV），2019：9196-9205.
[24] HOU Q B，ZHOU D Q，FENG J S.Coordinate attention for efficient mobile network design[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition，2021：13713-13722.