Improved Algorithm for Hot-Rolled Steel Strip Surface Defect Detection Based on YOLOv8
摘要
For the issue of low detection accuracy due to the large variation in the shape and size of defects on the surface of hot strip steel, an improved detection algorithm based on YOLOv8 is proposed. LSKNet is used to replace the original backbone network, and Dual Attention Block attention mechanism and Strip Block strip convolution module are introduced to enhance the feature extraction capability. Experiments show that the algorithm achieves a mAP of 77.5% on the NEU-DET dataset, which is 4.8% higher than the original algorithm.