- Research Article
- 10.1049/icp.2025.4646
Research on surface defect detection of strip steel based on attention mechanism
- Jan 01, 2026
- IET conference proceedings.
- P Wang + 7 more +7
This study aims to enhance the accuracy and robustness of surface defect detection in steel strips. Addressing the challenges of complex defect morphology, low detection accuracy, and high false alarm rates in existing methods, this paper introduces deep learning techniques. It optimizes the YOLOv7 algorithm, proposing multiple improvement strategies. Specifically, these include: employing deformable convolutions (DConv) to address morphological variations in defects; designing a bidirectional cross-attention feature fusion module (BCC) to leverage information across different feature layers fully; constructing a decoupled detection head (TSCODE) to optimize classification and localization tasks; utilizing the SIoU loss function to accelerate bounding box regression; and applying the K-means++ algorithm to obtain predefined anchor boxes. Experimental results demonstrate that the optimized algorithm achieves a mean average precision (mAP) of 80.1% on the strip steel surface defect dataset, representing a 6.0% improvement over the original method. The proposed optimization strategies significantly enhance the performance of strip steel surface defect detection.
Read more