Abstract:Steel surface defect detection is an indispensable part of industrial production and processing processes. It helps to reduce production costs, ensure product quality, improve production safety and compliance, and maintain sustainability and competitiveness. To address the low detection accuracy of traditional methods, this paper developed and investigated an improved algorithm based on YOLO for steel surface defect detection: ECA(Efficient Channel Attention) -SimSPPF (Simplified Spatial Pooling - Fast) -SIoU (Scylla Intersection over Union) -Yolov5. First, deformable convolutions were used to replace some conventional convolutions in the model, which expanded the receptive field and improved detection accuracy. Additionally, Efficient Channel Attention was integrated into the model to improve the weight of important information. Then, the SimSPPF was employed in place of the SPP module in the model, reducing computational complexity. Finally, the SIoU loss function was utilized to handle bounding box regression more effectively. The paper conducted different ablation experiments, and the improved ECA-SimSPPF-SIoU-Yolov5 algorithm demonstrated superior detection performance. Using the NEU-DET dataset, the mAP reached 78.8%, which was a 7.1% improvement higher than the original model, while the Recall reached 76.4% and improved by 3.7% compared to the original model. The improved model showed significant improvements in terms of mAP and Recall. Furthermore, the paper conducted multiple comparative experiments, comparing the model with other attention mechanisms and loss functions. The results demonstrated that the improved ECA-SimSPPF-SIoU-Yolov5 algorithm achieved good detection results in terms of mAP and Recall. In the third comparative experiment, the model was compared with YOLOv5 model with different network depths and the latest Yolov8 model, and the improved model also achieved good detection accuracy.

Steel Surface Defect Detection Based on Improved MASK RCNN

A deep-learning-based approach for fast and robust steel surface defects classification

A New Steel Defect Detection Algorithm Based on Deep Learning

Defective Surface Detection Based on Improved Faster R-CNN

A deep learning model for steel surface defect detection

Steel Surface Defect Detection via Deformable Convolution and Background Suppression

Multiscale Local and Global Feature Fusion for the Detection of Steel Surface Defects

Steel Surface Defect Detection Method Based on Improved YOLOv8

Robustness defect detection: Improving the performance of surface defect detection in interference environment

Surface defect detection of hot rolled steel based on multi-scale feature fusion and attention mechanism residual block

A hierarchical training-convolutional neural network with feature alignment for steel surface defect recognition

Steel Surface Defect Detection Using Improved Deep Learning Algorithm: ECA-SimSPPF-SIoU-Yolov5

A steel surface defect inspection approach towards smart industrial monitoring

Surface Defect Detection of Steel Strips Based on Anchor-Free Network With Channel Attention and Bidirectional Feature Fusion

Automated steel surface defect detection and classification using a new deep learning-based approach

Efficient Detection Model of Steel Strip Surface Defects Based on YOLO-V7

Defect Detection in Steel Using a Hybrid Attention Network

An End-to-End Steel Surface Defect Detection Approach via Fusing Multiple Hierarchical Features

Steel surface defect detection based on sparse global attention transformer

An Improved Mask R-CNN Micro-Crack Detection Model for the Surface of Metal Structural Parts

Research on Steel Surface Defect Detection Algorithm Based on Improved YOLOv8n