高级检索

改进YOLOv11-OBB的焊缝识别算法

Weld seam recognition algorithm based on improved YOLOv11-OBB

  • 摘要: 为了解决焊接自动化中焊缝粗定位面临的旋转目标、多尺度、大长宽比等问题,提出了一种基于改进YOLOv11旋转边界框(YOLOv11 oriented bounding box,YOLOv11-OBB)模型的焊缝粗定位方法.首先,设计了OBB-RLSCD检测头,以深度卷积与逐点卷积组合捕捉旋转特征,在轻量化架构下精准捕捉旋转特征,提升检测效率与精度;其次,采用自适应轻量化下采样模块改进卷积,在降低计算量的同时有效保留焊缝关键特征;另外,在C3k2模块中集成动态卷积,使其能依据输入特征自适应调整卷积操作;最后,引入PIoU损失函数,从像素级优化旋转目标定位精度,增强复杂场景适应性.结果表明,模型平均精度均值达到86.5%,较原始YOLOv11-OBB模型提升3.9%,参数量降低10.4%,计算量降低18.2%,推理速度达89.7帧/秒,满足焊接机器人实时粗定位需求.同时在公开数据集NEU-DET上进行验证,模型仍能保持较高的检测精度,证明了算法在不同工业场景下具备良好的适应性.

     

    Abstract: To address the problems of rotated objects, multiple scales, and large aspect ratios encountered in the coarse positioning of weld seams during welding automation, a coarse positioning method for weld seams based on an improved YOLOv11 oriented bounding box (YOLOv11-OBB) model was proposed. First, an OBB-RLSCD detection head was designed to capture rotational features using a combination of depthwise convolution and pointwise convolution, accurately capturing rotational features under a lightweight architecture and improving the detection efficiency and accuracy. Second, an adaptive lightweight downsampling module was adopted to improve the convolution, effectively retaining the key features of weld seams while reducing the amount of computation. Additionally, dynamic convolution was integrated into the C3k2 module to enable the convolution operation to adaptively adjust according to the input features. Finally, the PIoU loss function was introduced to optimize the positioning accuracy of rotated objects at the pixel level, enhancing the adaptability to complex scenes. The results indicate that the mean average precision of the model reaches 86.5%, which is 3.9% higher than that of the original YOLOv11-OBB model; the number of parameters is reduced by 10.4%; the computation amount is reduced by 18.2%, and the inference speed reaches 89.7 frames/s, which meets the demand for real-time coarse positioning of welding robots. Meanwhile, the model is validated on the public dataset NEU-DET, and it still maintains high detection accuracy, which proves that the algorithm has good adaptability in different industrial scenarios.

     

/

返回文章
返回