关键词:
焊接机器人
焊接缺陷检测
深度学习
RT-DETR
YOLOv8
摘要:
在焊接机器人进行焊接的过程中,受到操作不当或工艺控制不严等原因,往往会伴随焊接缺陷的产生。这些缺陷不仅会导致产品机械性能的下降和使用寿命的缩短,而且会诱发严重的工业事故,造成重大经济损失和安全隐患。因此,研究具备高精度识别能力的焊接机器人焊接缺陷智能检测模型,对减少焊接质量隐患、提升工业制造质量以及降低人工检测成本具有非常重要的理论和实际意义。
为此,本文基于深度学习网络构建了两个焊接机器人焊接缺陷检测模型,并通过对其网络结构的轻量化、二次改进等操作,使其模型检测性能得到了显著的提升。本文主要的工作和创新点如下:
(1)收集了焊接缺陷数据集并对数据集进行了预处理。首先,在数据集收集阶段,通过文献查阅法和网页搜索法,收集了最具代表的焊接缺陷数据集,用以内、外部焊接缺陷检测。其次,针对收集的开源数据集呈现显著的类别不平衡特性,对原始的焊接图像数据集实施了系统的图像数据增强策略,以此缓解了焊接数据不平衡所带来的模型训练过拟合、泛化能力弱化等问题。
(2)针对焊接机器人外部缺陷检测中存在的小目标缺陷检测漏检、误检以及缺陷角度任意分布的挑战,提出了一种基于YOLOv8(YOLO系列检测算法)的二次改进模型—YOLO-weld。首先,设计SPPF-weld和C2f_DBB_CBAM模块,用于增强YOLO-weld模型的上下文信息的聚合能力、多尺度特征的融合效果。然后引入OBB检测头用以准确地捕捉有方向性的缺陷,以此提高检测精度。最后,实验结果表明,对比YOLOv8模型,YOLO-weld模型在精度、召回率和mAP@0.5(Mean Average Precision at IoU=0.5)分别提升了3.2%、2.6%、4.0%和3.9%。
(3)针对焊接机器人焊缝内部缺陷检测中存在检测精度低和实时性差的问题,提出了RT-DETR(Real-Time Detection Transformer)的改进优化模型—LW-DETR。首先,通过剔除RT-DETR模型中冗余的模块和对其超参数进行调整,实现和满足了模型的轻量化和实时性的需求。其次,设计了基于空间协调注意力SCSA的改进模块-SCSA-EF,提升了对小目标的特征表达和模型泛化能力。接着,设计了基于大型可分离核注意力模块LSKA的改进模块-ELSKA,以提高模型检测大、中尺寸特征图的精度和鲁棒性。另外,添加损失函数Inner WioU为模型提供一种更细粒度的评估方式。最后,实验结果表明LW-DETR模型的全类别精度、召唤率、调和平均值和平均精度均值mAP@0.5和每秒帧数分别达97.9%、96.8%、97.0%、99.0%和34.2帧/s,分别比RT-DETR模型提升2.2%、0.7%、1.0%、1.2%、和175.8%,且计算量和参数量分别降低至RT-DETR模型的11.8%、18.3%。