高级检索

    基于GSL-YOLO的钢材表面缺陷检测算法

    Steel Surface Defect Detection Algorithm Based on GSL-YOLO

    • 摘要: 钢材在现代工业生产中因加工和环境因素易产生表面缺陷,影响其质量和使用寿命。针对现有检测方法存在的精度不足与计算复杂度较高的问题,本文提出了一种基于YOLOv8的改进算法GSL-YOLO,针对特征冗余和特征表达能力不足的问题,在C2f模块中引入Ghost Bottleneck以替代标准Darknet Bottleneck,减少冗余计算并提高特征提取效率,采用DynamicConv替换标准CBS模块,以增强对复杂场景的适应能力。为进一步提升模型对细小缺陷的检测精度,在颈部网络中引入SENetV1注意力机制,通过自适应调整通道权重,加强对关键特征的关注。此外,针对检测头参数量大、计算复杂度高的问题,设计轻量级共享卷积检测头LSDECD,有效降低模型参数量和计算开销。实验结果表明,GSL-YOLO在NEU-DET钢材表面缺陷数据集上的mAP50较YOLOv8提升3%,同时参数量减少33%,计算成本降低37%。

       

      Abstract: Steel is susceptible to surface defects caused by processing and environmental factors in modern industrial production, which adversely affect its overall quality and service life. To overcome the shortcomings of existing detection methods, such as inadequate accuracy and high computational complexity, this paper proposes an improved YOLOv8-based algorithm called GSL-YOLO. Specifically, Ghost Bottleneck is introduced in the C2f module to replace the standard Darknet Bottleneck, thereby reducing redundant computation and enhancing feature extraction efficiency. DynamicConv is then adopted in place of the standard CBS module to better adapt to complex scenarios. In order to further improve the detection accuracy for minute defects, SENetV1 is integrated into the neck network, enabling adaptive channel weighting that strengthens attention to key features. In addition, a lightweight shared convolution detection head named LSDECD is designed to address the large parameter volume and computational complexity of existing detection heads, effectively reducing both model parameters and computational overhead. Experimental results demonstrate that, on the NEU-DET steel surface defect dataset, GSL-YOLO achieves a 3 percentage-point increase in mAP50 compared with YOLOv8, while simultaneously reducing the number of parameters by 33% and lowering computational cost by 37%. This work not only improves detection accuracy but also provides a lightweight design that satisfies the requirements of real-time industrial detection.

       

    /

    返回文章
    返回