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

Abstract

During the production process of steel, there are often some defects on the surface of the product. Therefore, detecting defects is the key to produce high-quality products. At the same time, the defects of the steel have caused huge losses to the high-tech industry. A steel surface defect detection algorithm based on improved YOLO-V7 is proposed to address the problems of low detection speed and low detection accuracy of traditional steel surface defect detection methods. First, we use the de-weighted BiFPN structure to make full use of the feature information to strengthen feature fusion, reduce the loss of feature information during the convolution process, and improve the detection accuracy. Secondly, the ECA attention mechanism is combined in the backbone part to strengthen the important feature channels. Finally, the original bounding box loss function is replaced by the SIoU loss function, where the penalty term is redefined by taking the vector angle between the required regressions into account. The experimental results show that the improved model proposed in this paper has higher performance compared with other comparison models. Based on our experiments, the proposed method yields 80.2% mAP and 81.9% on the GC10-DET dataset and NEU-DET dataset with high speed, which is better than other existing models.

Keywords

• Machine vision
• object detection
• deep learning
• feature extraction