[发明专利]焊缝缺陷检测方法、装置、可读存储介质及电子设备

说明书

本申请提供一种焊缝缺陷检测方法、装置、可读存储介质及电子设备，涉及工业制造技术领域，该方法包括：获取包含目标工件表面焊缝的目标图像；利用分割模型对目标图像进行图像分割得到表面焊缝图像，以及利用检测模型从目标图像中提取焊缝的多个候选检测框；将表面焊缝图像与多个候选检测框进行融合，得到目标融合图像；通过自注意力机制从目标融合图像中提取出目标特征图像，并基于目标特征图像判断目标图像中是否存在焊缝缺陷。本申请提供的焊缝缺陷检测方法、装置、可读存储介质及电子设备，用于解决相关技术中通过人工目视检测焊缝缺陷准确率较低的问题，提高工件焊缝缺陷的检测精度。

技术领域

本申请涉及工业制造技术领域，尤其涉及一种焊缝缺陷检测方法、装置、可读存储介质及电子设备。

背景技术

在工业制造技术领域，在对工件进行处理时，通常会涉及焊接工序。例如，将工件的两个部件进行焊接，焊接完成后，通常需要对焊缝的质量进行检测，以剔除质量不合格的产品，从而提高产品良率。

在对焊缝的质量进行检测时，主要是对焊缝是否存在缺陷进行检测。在相关技术中，通常是采用人工检测的方式对焊缝是否存在缺陷进行检测。

然而，该种焊缝检测方式受限于检测人员的专业水平，可能存在焊缝缺陷漏检、错检的情况出现，焊缝检测的准确率较低，进而影响产品的合格率。

发明内容

本申请的目的是提供一种焊缝缺陷检测方法、装置、可读存储介质及电子设备，用于解决相关技术中通过人工目视检测焊缝缺陷准确率较低的问题，提高工件焊缝缺陷的检测精度。

本申请提供一种焊缝缺陷检测方法，包括：

获取包含目标工件表面焊缝的目标图像；利用分割模型对所述目标图像进行图像分割得到表面焊缝图像，以及利用检测模型从所述目标图像中提取焊缝的多个候选检测框；将所述表面焊缝图像与所述多个候选检测框进行融合，得到目标融合图像；通过自注意力机制从所述目标融合图像中提取出目标特征图像，并基于所述目标特征图像判断所述目标图像中是否存在焊缝缺陷。

可选地，所述利用分割模型对所述目标图像进行图像分割得到表面焊缝图像，包括：将所述目标图像转换为HSV图像，并通过对所述HSV图像的通道V的调整，拉伸所述HSV图像的亮度，并得到亮度提升图像；将所述亮度提升图像转换为RGB图像，并利用所述焊缝边界提取单元的特征提取模型对所述RGB图像进行特征提取，得到图像金字塔。

可选地，所述将所述亮度提升图像转换为RGB图像，并利用所述焊缝边界提取单元的特征提取模型对所述RGB图像进行特征提取，得到图像金字塔之后，所述方法还包括：将所述图像金字塔中的各个层级对应的第一特征图像进行自适应平均池化处理，并通过所述焊缝边界提取单元的全卷积模型对处理后的各个层级的特征图像进行特征提取，得到各个层级对应的第二特征图像；对各个层级对应的第二特征图像进行上采样处理，将每个层级对应的第二特征图像的尺寸调整为目标尺寸，并将每个层级对应的特征图像进行特征融合，得到全局特征图像；将所述全局特征图像进行归一化处理，得到0,1分布的第一识别结果图像；0值表示非焊缝区域，1值表示焊缝区域；基于所述第一识别结果图像中各个区域的数值，得到所述表面焊缝图像。

可选地，所述利用检测模型从所述目标图像中提取焊缝的多个候选检测框，包括：利用特征提取模型提取从所述目标图像中提取出第三特征图像，并将所述第三特征图像输入到区域候选网络模型中，提取所述多个候选检测框。

可选地，所述将所述表面焊缝图像与所述多个候选检测框进行融合，得到目标融合图像，包括：将所述表面焊缝图像以区别于所述多个候选检测框的RGB通道的第四通道与所述多个候选检测框进行图像融合，得到所述目标融合图像。