基于改进YOLOv8-Seg模型的电解铜表面跨尺度缺陷检测技术研究

Research on Cross-Scale Defect Detection Technology for Electrolytic Copper Surface based on an Improved Yolov8-Seg Model

  • 摘要: 针对现有深度学习表面缺陷检测模型结构复杂、跨尺度目标检测效果差的问题,以阴极铜板表面缺陷为检测对象,提出了一种基于YOLOv8-Seg改进的缺陷检测模型。首先,引入CBAM注意力机制,减少通道注意和空间注意对无关背景信息的干扰。然后,在模型的主干结构中用GELAN模块代替C2f模块,增强了模型对小目标的检测能力。接着,在解决梯度消失问题的基础上,将激活函数修改为GELU函数,提高模型检测的准确率和召回率,加速模型收敛。最后,为了满足工业生产可视化需要及评估分散的缺陷信息,对缺陷的局部图像依次进行灰度变换、高斯滤波去噪、逆二值化和缺陷轮廓标注,计算出缺陷区域面积。实验结果表明,改进后的模型在自建阴极铜板表面缺陷数据集中的检测准确率和召回率较原始模型分别提高了6.5%和11.5%,在东北大学钢板表面缺陷数据集和天池铝型材表面缺陷数据集中的检测准确率分别达到了92.8%和74.3%,验证了改进方法的有效性。

     

    Abstract: To address the problems of complex architecture and poor cross-scale target detection performance in existing deep learning surface defect detection models, a defect detection model based on the YOLOv8-Seg improvement is proposed, with a focus on detecting surface defects in cathode copper plates. Firstly, the CBAM attention mechanism is introduced to reduce the interference of channel and spatial attention on irrelevant background information. Next, the GELAN module is used to replace the C2f module in the backbone structure of the model, enhancing the model’s ability to detect small targets. Then, to solve the gradient vanishing problem, the activation function is modified to GELU, improving the model’s detection accuracy and recall rate, while accelerating convergence. Finally, to meet the visualization needs of industrial production and assess scattered defect information, local images of defects are sequentially processed using grayscale transformation, Gaussian filtering for denoising, inverse binarization, and defect contour labeling, followed by the calculation of the defect area. Experimental results show that the improved model achieved a 6.5% and 11.5% increase in detection accuracy and recall rate, respectively, compared to the original model on the self-built cathode copper plate surface defect dataset. The detection accuracy reached 92.8% and 74.3% on the Northeastern University steel plate surface defect dataset and the Tianchi aluminum profile surface defect dataset, respectively, validating the effectiveness of the proposed improvement.

     

/

返回文章
返回