An Algorithm for Detecting Surface Defects in Steel Strips Based on an Improved Lightweight Network

Da Zhan,Han Wang,Xiao Yang,Wei Ou,Ren Huang,Jian Lin,K. Yi,Bei Zhou
DOI: https://doi.org/10.4028/p-foi56w
2024-01-01
Abstract:In recent years, surface defect detection methods based on deep learning have been widely applied to steel plate surface defect detection. By locating and classifying defects on the surface of steel plates, production efficiency can be improved. However, there is still a conflict between speed and accuracy in the defect detection process. To address this issue, we propose a high-precision, low-latency surface defect detection algorithm called the GhostConv-ECA-YOLOv5 Network (GEA-Net). The GEA-Net model can predict defect categories without compromising classification and detection accuracy. Experimental results show that our proposed improved model has higher performance compared to other comparative models, achieving a 75.6% mAP on the NEU-DET dataset.
What problem does this paper attempt to address?