YOLOv8-LGCV:面向热轧产线随线检测的轻量化钢材表面缺陷检测模型

YOLOv8-LGCV: Lightweight steel surface defect detection model for in-line inspection of hot rolling production line

  • 摘要: 针对钢材表面缺陷检测模型参数量大、计算开销高及小目标缺陷检测精度不足等问题,以YOLOv8为基线,提出了一种轻量化高精度的钢材表面缺陷检测模型。在主干网络中将C2f模块替换为基于Ghost卷积的C2f_Ghost模块,利用主特征加“幻影”特征生成机制将总参数量减少约16.7%;在颈部特征融合网络中引入VoVGSCSP模块,融合GSConv通道混洗与CSP跨阶段连接结构,将颈部计算量降低约35%,同时,在颈部网络进行通道的轻量化操作;在颈部P3以及P4输出层嵌入坐标注意力模块,增强对划痕、轧制氧化铁皮等方向性缺陷的检测能力。在东北大学NEU-DET钢材表面缺陷数据集上的实验结果表明,所提模型的mAP@0.5达到77.4%,计算量为5.9 GFLOPs,参数量为2.2 M,权重文件仅4.6 MB,在NVIDIA Jetson Orin Nano边缘推理终端上的实机测试表明,模型推理速度达79.6 f/s,单帧推理延迟约12.6 ms,在推理时延、功耗与边缘端资源占用等实时性指标上满足热轧产线随线检测的边缘部署要求。需说明的是,本文精度指标在NEU-DET等公开基准数据集上获得,面向真实产线图像的检测精度验证将在后续工作中开展。

     

    Abstract: To address the issues of large parameter count, high computational overhead, and insufficient detection accuracy for small-target defects in steel surface defect detection models, a lightweight and high-precision steel surface defect detection model is proposed based on the YOLOv8 baseline. In the backbone network, the C2f module is replaced with the C2f_Ghost module based on Ghost convolution, which leverages the generation mechanism of primary features plus "ghost" features to reduce the total number of parameters by approximately 16.7%. In the neck feature fusion network, the VoVGSCSP module is introduced, integrating the GSConv channel-shuffle mechanism with CSP cross-stage partial connections, which reduces the computational cost of the neck by approximately 35%. Meanwhile, channel lightweighting operations are performed on the neck network. Additionally, Coordinate Attention modules are embedded in the P3 and P4 output layers of the neck to enhance the detection capability for directional defects such as scratches and rolled-in scale. Experimental results on the NEU-DET steel surface defect dataset from Northeastern University show that the proposed model achieves an mAP@0.5 of 77.4%, with 5.9 GFLOPs, 2.2M parameters, and a weight file size of only 4.6 MB. Real-machine tests on an NVIDIA Jetson Orin Nano edge inference device showed that the model achieved an inference speed of 79.6 frames/s and a single-frame latency of approximately 12.6 ms. In terms of real-time performance indicators, including inference latency, power consumption, and edge-device resource utilization, the model meets the requirements for edge deployment in in-line inspection on hot-rolling production lines. It should be noted that the accuracy metrics reported in this study were obtained on public benchmark datasets such as NEU-DET; the model’s detection performance on images collected from actual production lines remains to be further validated.

     

/

返回文章
返回