Article
Version 1
This version is not peer-reviewed
YOLO-LFPD: A Lightweight Method for Strip Surface Defect Detection
Version 1
: Received: 9 July 2024 / Approved: 9 July 2024 / Online: 10 July 2024 (11:50:29 CEST)
How to cite: Lu, J.; Zhu, M.; Qin, K.; Ma, X. YOLO-LFPD: A Lightweight Method for Strip Surface Defect Detection. Preprints 2024, 2024070783. https://doi.org/10.20944/preprints202407.0783.v1 Lu, J.; Zhu, M.; Qin, K.; Ma, X. YOLO-LFPD: A Lightweight Method for Strip Surface Defect Detection. Preprints 2024, 2024070783. https://doi.org/10.20944/preprints202407.0783.v1
Abstract
Strip steel surface defect recognition research has important research significance in industrial production. Aiming at the problems of defect feature extraction difficulty, slow detection speed and insufficient dataset, YOLOv5 is improved on the basis of YOLOv5, and YOLO-LFPD (Lightweight fine particle detection) model is proposed. By introducing the RepVGG (Re-param VGG) module, the robustness of the model to the surface defect features of strip steel is enhanced, and the expressive ability of the model is improved. FasterNet is used to replace the backbone network, which ensures the accuracy and accelerates the inference speed, making the model more suitable for real-time monitoring. The use of pruning, GA genetic algorithm with OTA loss function method further reduces the model size while better learning the strip steel defect feature information, thus improving the generalisation ability and accuracy of the model. The experimental results show that the introduction of the RepVGG module and the use of FasterNet can well improve the model performance, with a reduction of 48% in the number of parameters, a reduction of 13% in the number of GFLOPs, an inference time of 77% of the original, and an optimal accuracy compared with the network models in recent years. The experimental results on the NEU-DET dataset show that the accuracy of YOLO-LFPD is improved by 3% to 81.2%, which is better than other models, and provides new ideas and references for the lightweight strip steel surface defect detection scenarios and application deployment.
Keywords
Surface defect recognition; Lightweighting; YOLO; Pruning; FasterNet
Subject
Computer Science and Mathematics, Computer Vision and Graphics
Copyright: This is an open access article distributed under the Creative Commons Attribution License which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
Comments (0)
We encourage comments and feedback from a broad range of readers. See criteria for comments and our Diversity statement.
Leave a public commentSend a private comment to the author(s)
* All users must log in before leaving a comment
