基于YOLOv5s-FCS的钢材表面缺陷检测  

Steel Surface Defect Detection Based on YOLOv5s-FCS

作　　者：周孟然[1] 王昊男 高立鹏 王宁 来文豪 ZHOU Meng-ran;WANG Hao-nan;GAO Li-peng;WANG Ning;LAI Wen-hao(College of Electrical and Information Engineering,Anhui University of Science and Technology,Huainan 232000,China;College of Mechanics and Optoelectronic Physics,Anhui University of Science and Technology,Huainan 232000,China)

机构地区：[1]安徽理工大学电气与信息工程学院,淮南232000 [2]安徽理工大学力学与光电物理学院,淮南232000

出　　处：《科学技术与工程》2024年第14期5901-5910,共10页Science Technology and Engineering

基　　金：安徽省科技重大专项(201903a07020013);安徽理工大学2022年博士创新基金(2022CX1007)。

摘　　要：针对传统钢材表面缺陷检测方法易出现误检、漏检和部分缺陷种类检测精度低等问题,设计了一种钢材表面缺陷网络YOLOv5s-FCS。首先引用了FReLU激活函数构建了卷积模块CBF,有效增强了网络的空间解析能力,优化了网络检测精度;其次,将坐标注意力机制嵌入到网络的neck部分来增强网络特征融合的能力,从而使网络能够提取更加丰富的特征信息;最后,将YOLOv5s的损失函数替换为SIoU loss,提高了预测框的回归精度。通过在NEU-DET数据集上进行消融实验、可视化对比实验,结果表明,YOLOv5s-FCS网络的mAP值达到了0.747,相较于原YOLOv5s网络提高了8.3%,相较于YOLOv3网络提高了11.8%,相较于YOLOXs网络提高了4.2%,相较于YOLOv6s提高了1.4%,验证了该方法的可行性、有效性。The YOLOv5s-FCS network for traditional steel materials,which addresses issues such as false positives,false negatives,and low accuracy in detecting certain types of defects was presented.Firstly,the CBF convolution module was constructed using the FReLU activation function to enhance the network's spatial resolution capability and optimize detection accuracy.Secondly,a coordinate attention mechanism was embedded into the neck part of the network to enhance its feature fusion capability,enabling the extraction of more rich feature information.Finally,the SIoU loss replaces the YOLOv5s loss function to improve the regression accuracy of the predicted box.Through ablation experiments and visualization comparisons on the NEU-DET dataset,it is demonstrated that the mAP value of the YOLOv5s-FCS network reaches 0.747,representing an improvement of 8.3%compared to the original YOLOv5s network,11.8%compared to the YOLOv3 network,4.2%compared to the YOLOXs network,and 1.4%compared to the YOLOv6s network,thus demonstrating the feasibility and effectiveness of the proposed method.

关 键 词：钢材 缺陷检测 深度学习 注意力机制 

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