Surface defect detection of steel strips based on improved YOLOv4

Abstract

During the production and processing of steel strips, the production process and external factors lead to surface defects that negatively impact the strips’ integrity and functionality. However, traditional manual defect detection algorithms cannot meet modern accuracy requirements. Therefore, we propose a steel strip surface defect detection method based on the improved you-only-look-once version 4 (YOLOv4) algorithm. The attention mechanism is embedded in the backbone network structure, and the path aggregation network is modified into a customised receptive field block structure, which strengthens the feature extraction functionality of the network model. From the final experimental results, relative to the original YOLOv4 algorithm, the proposed algorithm's mean average precision values in the detection of four types of steel strip defects is improved by 3.87%, reaching 85.41%, thereby providing a new detection method for daily steel strip surface defects.