改进YOLOv5的安全帽佩戴检测算法  

Improved HelmetWear Detection Algorithm for YOLOv5

作　　者：乔炎 甄彤[1,2] 李智慧 QIAO Yan;ZHEN Tong;LI Zhihui(College of Information Science and Engineering,Henan University of Technology,Zhengzhou 450001,China;Key Laboratory of Grain Information Processing and Control(Henan University of Technology),Ministry of Education,Zhengzhou 450001,China)

机构地区：[1]河南工业大学信息科学与工程学院,郑州450001 [2]粮食信息处理与控制重点实验室,郑州450001

出　　处：《计算机工程与应用》2023年第11期203-211,共9页Computer Engineering and Applications

基　　金：河南省“揭榜挂帅”项目(211110110500)。

摘　　要：针对目前目标检测模型结构复杂、计算量大、检测准确率低等问题,提出在工业场景下基于改进型YOLOv5的安全帽佩戴算法。在主干网络引入轻量型网络ShuffleNetv2,保留Focus结构和ShuffleNetv2共同组成主干网络,降低网络的计算量和参数量;在C3模块中引入Swin Transformer Block,得到C3STB模块,替换Neck部分原有的C3模块;设计了CBAM_H注意力机制,并将其嵌入Neck网络中,获取全局上下文信息,提高模型检测准确率。自建数据集并进行实验,实验结果表明,改进后的YOLOv5模型的参数量由6.14×10^(6)压缩到8.9×10^(5),计算量由1.64×10^(10)压缩到6.2×10^(9),mAP由0.899上升到0.908,优于原模型性能。Aiming at the problems of complex structure,large computation and low detection accuracy of the current target detection model,the helmet wearing algorithm based on the improved YOLOv5 is proposed in industrial scenarios.Firstly,the light-weight network ShuffleNetv2 is introduced in the backbone network,and the Focus structure and Shuffle-Netv2 are retained to jointly form the backbone network to reduce the computation and number of parameters of the network;secondly,the Swin Transformer Block is made to be introduced in the C3 module to obtain the C3STB module,replacing the original C3 module in the Neck part;finally,the CBAM_H attention mechanism is designed and embedded in the Neck network to obtain global context information and improve the model detection accuracy.The experimental results show that the improved YOLOv5 model compresses the number of parameters from 6.14×10^(6) to 8.9×10^(5),the computational volume from 1.64×10^(10) to 6.2×10^(9),and the mAP from 0.899 to 0.908,which is better than the performance of the original model.

关 键 词：YOLOv5 ShuffleNetv2 CBAM注意力机制 Swin Transformer Block 

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