高级检索

改进YOLO11焊缝表面缺陷检测模型

Improved YOLO11 model for weld surface defect detection

  • 摘要: 针对复杂背景中焊接缺陷尺度变化大、对比度低,缺陷多样化重叠问题,提出一种改进YOLO11焊缝表面缺陷检测算法. 主干网络引入特征提取DWR(dilation-wise residual)模块,并与C3k2模块融合,增强模型信息的提取能力;引入LSKA(large separable kernel attention)注意力模块,对多尺度特征融合SPPF模块改进,使模型关注主要特征;采用Inner-SIoU损失函数,提高边框回归精度. 结果表明,改进模型能快速准确检测缺陷,其精度、召回率和PmAP50达到92.5%,92.2%和95.3%,比基准模型高2.1%,3.0%和3.5%,模型参数量和浮点运算量分别减少了0.127 M和0.3 G,推理速度达293帧/s,表明改进模型显著提升,推理速度快,有效降低误检与漏检率,尤其在小尺寸缺陷检测方面表现突出,兼具高精度与轻量化优势,更适合工业场景下的高效焊缝缺陷检测应用.

     

    Abstract: To address the problems of large scale variation, low contrast, and diverse overlapping of welding defects in complex backgrounds, an improved YOLO11 algorithm for weld surface defect detection was proposed. A feature extraction dilation-wise residual (DWR) module was introduced into the backbone network and fused with the C3k2 module to enhance the information extraction capability of the model. A large separable kernel attention (LSKA) module was introduced to improve the multi-scale feature fusion SPPF module, enabling the model to focus on primary features. The Inner-SIoU loss function was adopted to improve the bounding box regression accuracy. The results indicate that the improved model can detect defects quickly and accurately. Its precision, recall, and PmAP50 reach 92.5%, 92.2%, and 95.3%, respectively, which are 2.1%, 3.0%, and 3.5% higher than those of the baseline model. The model parameters and floating-point operations are reduced by 0.127 M and 0.3 G, respectively, and the inference speed reaches 293 frames/s. This indicates that the improved model is significantly enhanced with fast inference speed, effectively reducing the false detection and missed detection rates. It performs particularly well in small-sized defect detection, possesses the advantages of both high accuracy and lightweight, and is more suitable for efficient weld defect detection applications in industrial scenarios.

     

/

返回文章
返回