融合坐标注意力的YOLOv8焊接缺陷检测方法

Coordinate Attention-enhanced YOLOv8 for Welding Defect Detection

  • 摘要: 改进YOLOv8检测算法。在数据增强环节,通过Mixup结合随机旋转的数据增强策略,缓解稀有缺陷类别样本数量不足的 问题。网络改进方面,把YOLOv8骨干网络交叉阶段部分模块替换为坐标注意力交叉阶段部分模块,以此强化缺陷位置敏 感检测能力,用渐进特征金字塔网络头代替标准检测头,提升多尺度特征融合效果。后续,本文使用开源焊接缺陷数据集 GC10-DET进行相关对比试验,试验结果显示方法在检测精度和小缺陷召回率方面都有提高,为焊接零部件缺陷检测提供 了高精度和高鲁棒性的解决办法。

     

    Abstract: detection,an improved YOLOv8 detection algorithm integrating coordinate attention and asymptotic feature pyramid network is proposed. In the data augmentation stage,a combination of Mixup and random rotation is employed to alleviate the problem of insufficient samples of rare defect categories. For network improvement,certain modules in the cross-stage sections of the YOLOv8 backbone network are replaced with coordinate attention-based modules to enhance the sensitivity to defect location. Additionally, the detection head is replaced with a asymptotic feature pyramid network head to improve multi-scale feature fusion. Subsequently, comparative experiments are conducted using the open-source welding defect dataset GC10-DET. The experimental results show that the proposed method achieves improvements in both detection accuracy and recall rate for small defects,providing a high-precision and high-robustness solution for welding component defect detection.

     

/

返回文章
返回