高级检索

改进YOLOv8n的焊接缺陷检测方法

Improved YOLOv8n welding defect detection method

  • 摘要: 针对焊接缺陷具有多尺度,形态复杂和易受背景干扰等特点,提出一种基于YOLOv8n的焊接缺陷检测算法YOLO-SBRS. 首先利用空间和通道重建卷积(Spatial and Channel Reconstruction Convolution,SCConv)卷积改进主干网络的的 C2f模块;同时设计一种具有双层路由注意力机制的空间金字塔快速平均池化(Spatial Pyramid Pooling Fast Average Pooling with BiFormer Attention Module,SPPF_ABF)模块,将原始SPPF模块的最大池化替换为平均池化操作,并引入双层路由 Transformer 注意力机制;其次采用重参数化广义特征金字塔网络(Reparameterized Generalized Feature Pyramid Network,RepGFPN)优化特征融合部分;最后,基于参数共享原理及引入联合空间到深度层和非跨步卷积层模块(A Module Combining Space-To-Depth and Non-Strided Convolutional Layers,SPD_Conv)改进检测头,实现轻量化的同时提升网络对复杂缺陷的检测能力. 实验结果表明,改进后算法的精度和交并比为50%的平均精度均值(mean Average Precision at 50% Intersection over Union,mAP50)分别提高3.1%和2.8%,为焊接缺陷检测提供一种高效且可行的解决方案.

     

    Abstract: A welding defect detection algorithm YOLO-SBRS based on YOLOv8n is proposed for welding defects with characteristics of multi-scale, complex shape and vulnerable to background interference. First, Spatial and Channel Reconstruction Convolution(SCConv) is used to improve the C2f module of the backbone network; At the same time, a Spatial Pyramid Pooling Fast Average Pooling with BiFormer Attention Module (SPPF_ABF) is designed, which replaces the maximum pooling of the original SPPF module with the average pooling operation, and introduces a two-layer routing transformer attention mechanism; Secondly, a Reparameterized Generalized Feature Pyramid Network (RepGFPN) is used to optimize the feature fusion part; Finally, based on the principle of parameter sharing and the introduction of a Module Combining Space-To-Depth and Non-Strided Convolutional Layers (SPD_Conv), the detection head is improved to achieve lightweight and improve the detection capability of the network for complex defects. Experimental results show that the accuracy of the improved algorithm and mean Average Precision at 50% Intersection over Union (mAP50) are improved by 3.1% and 2.8% respectively,which provide an efficient and feasible solution for welding defect detection.

     

/

返回文章
返回