27 November 2024 Lightweight strip steel surface defect detection algorithm based on YOLOv8-VRLG
Hao Zhou, Yongping Zhang, Cheng Yan
Author Affiliations +
Abstract

Accurate detection of surface defects is crucial for maintaining the quality of strip steel products. We propose a lightweight strip steel surface defect detection model, YOLOv8-VRLG, to address the issues of insufficient detection accuracy and large parameter sizes in existing models. First, the VanillaNet module was integrated into YOLOv8 to preserve efficient feature extraction capabilities while effectively lightening the backbone network. Second, the C2fRL module, which integrates the benefits of RepGhost and large selective kernel modules into the original C2f, was introduced to further reduce the number of parameters and dynamically adjust processing details based on image content, thereby enhancing detection accuracy. Finally, the traditional convolutional layers in the detection head were replaced with the GSConv module. This module optimizes information exchange and feature fusion with its unique mixed convolution strategy, thereby improving the model’s accuracy and computational resource efficiency. Extensive tests on the NEU-DET dataset demonstrated that the mean average precision of the YOLOv8-VRLG model improved by 1.8% compared with that of the YOLOv8n model. In addition, the computational cost and parameter count were reduced by 23.5% and 28.6%, respectively. Tests on the GC10-DET dataset further showcased the model’s exceptional adaptability and robustness in complex environments.

© 2024 SPIE and IS&T
Hao Zhou, Yongping Zhang, and Cheng Yan "Lightweight strip steel surface defect detection algorithm based on YOLOv8-VRLG," Journal of Electronic Imaging 33(6), 063033 (27 November 2024). https://doi.org/10.1117/1.JEI.33.6.063033
Received: 13 August 2024; Accepted: 14 November 2024; Published: 27 November 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
Back to Top