Attention-guided YOLOv5s-SDF model for accurate detection of strip steel surface defects
摘要
Surface defects on strip steel are often inevitable due to limitations in raw materials and manufacturing processes. To improve defect detection accuracy, we propose the YOLOv5s-SDF algorithm, which integrates ShuffleAttention in the neck and DyHead in the head of the YOLOv5s framework. The novelty lies in the use of ShuffleAttention to model spatial-channel dependencies and the incorporation of Focal-CIoU loss to mitigate the influence of low-quality samples. Experiments on the NEU-DET dataset show that YOLOv5s-SDF achieves 69.37% Precision, 74.21% Recall, 76.32%