Steel Surface Defect Detection Based on YOLOv5s-FCS

Meng-ran ZHOU , Hao-nan WANG , Li-peng GAO , Ning WANG , Wen-hao LAI

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (14) : 5901 -5910.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (14) : 5901-5910 DOI: 10.12404/j.issn.1671-1815.2302837
research-article
Steel Surface Defect Detection Based on YOLOv5s-FCS
Author information +
History +
PDF

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.

Cite this article

Download citation ▾
Meng-ran ZHOU, Hao-nan WANG, Li-peng GAO, Ning WANG, Wen-hao LAI. Steel Surface Defect Detection Based on YOLOv5s-FCS. Science Technology and Engineering, 2024, 24(14): 5901-5910 DOI:10.12404/j.issn.1671-1815.2302837

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (2704KB)

17

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉