A Highly Efficient and Lightweight Detection Method for Steel Surface Defect

Abstract

The detection of steel surface defects is of great significance to steel production. In order to better meet the requirements of accuracy, real-time, and lightweight model, this paper proposes a highly efficient and lightweight steel surface defect detection method based on YOLOv5n. Firstly, ODMobileNetV2 composed of MobileNetV2 and ODConv is used as the backbone to improve the defect feature extraction capability. Secondly, GSConv is utilized in the neck to achieve deep information fusion through channel concatenation and shuffling, enhancing the ability of feature fusion. Finally, this paper proposes a spatial-channel reconstruction block (SCRB) designed to suppress redundant features and improve the representation ability of defect features through feature separation and reconstruction. Experimental results show that this method achieves 84.1% mAP and 109 FPS on the NEU-DET dataset, and 72.9% mAP and 110.1 FPS on the GC10-DET dataset, enabling accurate and efficient detection. Furthermore, the number of parameters is only 5.04M, which has a significant lightweight advantage.