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.

ST-CA YOLOv5: Improved YOLOv5 Based on Swin Transformer and Coordinate Attention for Surface Defect Detection

Swin-Transformer -YOLOv5 for lightweight hot-rolled steel strips surface defect detection algorithm

YOLOv5-ACCOF Steel Surface Defect Detection Algorithm

CFM-YOLOv5:CFPNet moudle and muti-target prediction head incorporating YOLOv5 for metal surface defect detection

MSFT-YOLO: Improved YOLOv5 Based on Transformer for Detecting Defects of Steel Surface

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

Surface Defect Detection for Automated Tape Laying and Winding Based on Improved YOLOv5

Strip steel surface defect detection based on lightweight YOLOv5

Attention-based deep learning for chip-surface-defect detection

Surface defect detection of ceramic disc based on improved YOLOv5s

Improved YOLOv7-based steel surface defect detection algorithm

Surface Defect Detection of Industrial Parts Based on YOLOv5

Surface Defect Detection Algorithm Based on Feature-Enhanced YOLO

Surface defect detection of industrial components based on vision

A Novel YOLOv5_ES based on lightweight small object detection head for PCB surface defect detection

PCB-YOLO: An Improved Detection Algorithm of PCB Surface Defects Based on YOLOv5

SSA-YOLO: An Improved YOLO for Hot-Rolled Strip Steel Surface Defect Detection

Experimental Design of Steel Surface Defect Detection Based on MSFE-YOLO—An Improved YOLOV5 Algorithm with Multi-Scale Feature Extraction

Object Detection Algorithm for Surface Defects Based on a Novel YOLOv3 Model

Surface Defect Detection of Steel Strip at Low Resolution Based on SAC-YOLOv5