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

周孟然 ,  王昊男 ,  高立鹏 ,  王宁 ,  来文豪

科学技术与工程 ›› 2024, Vol. 24 ›› Issue (14) :5901-5910

PDF (2704KB)
科学技术与工程 ›› 2024, Vol. 24 ›› Issue (14) :5901-5910 DOI: 10.12404/j.issn.1671-1815.2302837
自动化技术、计算机技术

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

作者信息 +

Steel Surface Defect Detection Based on YOLOv5s-FCS

Author information +
文章历史 +

摘要

针对传统钢材表面缺陷检测方法易出现误检、漏检和部分缺陷种类检测精度低等问题,设计了一种钢材表面缺陷网络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%,验证了该方法的可行性、有效性。

Abstract

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.

引用本文

引用格式 ▾
周孟然, 王昊男, 高立鹏, 王宁, 来文豪. 基于YOLOv5s-FCS的钢材表面缺陷检测[J]. 科学技术与工程, 2024, 24(14): 5901-5910 DOI:10.12404/j.issn.1671-1815.2302837
ZHOU Meng-ran, WANG Hao-nan, GAO Li-peng, WANG Ning, LAI Wen-hao. Steel Surface Defect Detection Based on YOLOv5s-FCS[J]. Science Technology and Engineering, 2024, 24(14): 5901-5910 DOI:10.12404/j.issn.1671-1815.2302837

登录浏览全文

4963

注册一个新账户 忘记密码

参考文献

基金资助

安徽省科技重大专项(201903a07020013)

安徽理工大学2022年博士创新基金(2022CX1007)

PDF (2704KB)

33

访问

0

被引

详细

导航
相关文章

/

〈 〉