The Steel Surface Multiple Defect Detection and Size Measurement System Based on Improved YOLOv5

Abstract

In the process of steel production, the defects on the surface of steel will adversely affect the subsequent processing of a product. Accurate detection of such defects is the key to improve production efficiency and economic benefits. In this paper, an end-to-end steel surface defect detection and size measurement system based on the YOLOv5 model is designed. Firstly, in consideration of the defect location and direction correlation in the production process, a coordinate attention mechanism is added at the head of YOLOv5 to strengthen the spatial correlation of the steel surface and an adaptive anchor box generation method based on defect shape difference feature is proposed, which realizes the detection of three main types of defects on the Pytorch deep learning framework. Secondly, BiFPN is used to strengthen the feature fusion and a transformer encoder is added to improve the performance of detecting small defects. Thirdly, calculate the conversion ratio between the pixel and the actual size according to the standard reference specimen and obtain the actual size through the pixel statistics of the defect area to achieve pixel level size measurement. Finally, the steel surface defect detection and size measurement system are designed in this paper, which consist of various hardware, related measurement, and detection algorithms. According to the experimental results, the comprehensive defect detection accuracy of this method reaches 93.6%, of which the scratch detection accuracy reaches 95.7%. The detection speed reaches 133 fps and the defect size measurement accuracy reaches 0.5 mm. Experimental result shows that the defect detection and size measurement system designed in this paper can accurately detect and measure various industrial production defects and can be applied to the actual production process.