Research on Metallurgical Saw Blade Surface Defect Detection Algorithm Based on SC-YOLOv5

Abstract

Under the background of intelligent manufacturing, in order to solve the complex problems of manual detection of metallurgical saw blade defects in enterprises, such as real-time detection, false detection, and the detection model being too large to deploy, a study on a metallurgical saw blade surface defect detection algorithm based on SC-YOLOv5 is proposed. Firstly, the SC network is built by integrating coordinate attention (CA) into the Shufflenet-V2 network, and the backbone network of YOLOv5 is replaced by the SC network to improve detection accuracy. Then, the SIOU loss function is used in the YOLOv5 prediction layer to solve the angle problem between the prediction frame and the real frame. Finally, in order to ensure both accuracy and speed, lightweight convolution (GSConv) is used to replace the ordinary convolution module. The experimental results show that the mAP@0.5 of the improved YOLOv5 model is 88.5%, and the parameter is 31.1M. Compared with the original YOLOv5 model, the calculation amount is reduced by 56.36%, and the map value is increased by 0.021. In addition, the overall performance of the improved SC-YOLOv5 model is better than that of the SSD and YOLOv3 target detection models. This method not only ensures the high detection rate of the model, but also significantly reduces the complexity of the model and the amount of parameter calculation. It meets the needs of deploying mobile terminals and provides an effective reference direction for applications in enterprises.