安徽工程大学学报2024，Vol.39Issue(5)：48-56,9.
基于轻量化YOLOv5s的带钢表面缺陷检测算法
Strip Steel Surface Defect Detection Algorithm Based on Lightweight YOLOv5s
摘要
Abstract
The detection of defects on the surface of strip steel is very important for industrial production.Aiming at the problems of large number of parameters and large computation volume of strip steel sur-face defect detection algorithms in industrial scenarios,a lightweight GDSB-YOLOv5 algorithm is pro-posed for detecting defects on strip steel.Firstly,GhostNet is used as the backbone network to reduce the number of parameters and the computation amount;meanwhile,depth-separable convolution(DSConv)is introduced to further reduce the number of parameters;secondly,Squeeze-and-Excitation(SE)attention module is added after the three concatenation to effectively improve the extraction of tar-get features.Finally,the BiFPN weighted bidirectional pyramid structure is introduced to replace the FPN+PAN structure to improve the fusion efficiency of the feature map.The experimental results show that the average detection accuracy of GDSB-YOLOv5 is 78％,which is 1.3％higher than the original YOLOv5s algorithm,and the computational and parametric quantities are reduced by 51.2％and 26％,respectively.The detection algorithm achieves the lightweight of the algorithm while ensuring the aver-age detection accuracy.关键词
YOLOv5s算法/轻量化/特征融合/注意力模块/缺陷检测Key words
YOLOv5s/lightweight/feature fusion/attention module/defect detection分类
信息技术与安全科学引用本文复制引用
李晓孛,瞿成明,单传辉..基于轻量化YOLOv5s的带钢表面缺陷检测算法[J].安徽工程大学学报,2024,39(5):48-56,9.基金项目
安徽工程大学国家自然科学基金预研项目(Xjky2022046) （Xjky2022046）
