[发明专利]一种管道焊缝缺陷检测方法及相关装置

权利要求书

1.一种管道焊缝缺陷检测方法，其特征在于，包括：

获取辅助图片集和检测图片集；

将所述辅助图片集和所述检测图片集输入特征提取网络，得到所述辅助图片集对应的第一特征图和所述检测图片集对应的第二特征图；

使用区域建议网络对所述第一特征图和所述第二特征图进行处理，得到所述检测图片集中预测的缺陷部位；

确定所述辅助图片集中标注的缺陷部位与所述检测图片集中预测的缺陷部位的相似度；

根据所述相似度，输出检测结果。

2.根据权利要求1所述的方法，其特征在于，所述获取辅助图片集包括：

采集管道焊缝的多类缺陷图片；

标注所述多类缺陷图片中的缺陷部位；

对所述多类缺陷图片中的缺陷部位进行分类，得到所述多类缺陷图片中的缺陷部位的类别；

根据所述多类缺陷图片和所述多类缺陷图片中的缺陷部位的类别，生成所述辅助图片集。

3.根据权利要求1或2所述的方法，其特征在于，所述特征提取网络包括多个子网络，所述多个子网络的权重共享，所述多个子网络中的其中一个子网络用于提取所述检测图片集的特征，所述多个子网络中的其他子网络用于提取所述辅助图片集的特征。

4.根据权利要求3所述的方法，其特征在于，所述辅助图片集包括N个类别的图片，所述N个类别的图片中每个类别的图片包括K张图片，其中，N和K都为正整数，所述其他子网络中的每个子网络分别用于提取所述每个类别的图片的特征，并且所述每个类别的图片的特征为所述每个类别的图片包括的K张图片的平均特征。

5.根据权利要求1所述的方法，其特征在于，所述使用区域建议网络对所述第一特征图和所述第二特征图进行处理，得到所述检测图片集中预测的缺陷部位，包括：

根据所述第一特征图和所述第二特征图，计算相似度特征图；

根据所述相似度特征图，确定所述检测图片集中预测的缺陷部位。

6.根据权利要求1所述的方法，其特征在于，所述确定所述辅助图片集中标注的缺陷部位与所述检测图片集中预测的缺陷部位的相似度包括：

使用多关系检测器对所述辅助图片集中标注的缺陷部位和所述检测图片集中预测的缺陷部位进行相似度检测，其中，所述多关系检测器包括M个相似度检测模块，M为大于1的整数；

获取所述M个相似度检测模块检测得到的M个相似度；

根据所述M个相似度，确定所述辅助图片集中标注的缺陷部位与所述检测图片集中预测的缺陷部位的相似度。

7.根据权利要求6所述的方法，其特征在于，M为3，所述获取所述M个相似度检测模块检测得到的M个相似度包括：

获取第一相似度检测模块检测得到的全局相似度；

获取第二相似度检测模块检测得到的局部相似度；

获取第三相似度检测模块检测得到的块相似度。

8.一种管道焊缝缺陷检测装置，其特征在于，包括：

获取模块，用于获取辅助图片集和检测图片集；

特征提取模块，用于将所述辅助图片集和所述检测图片集输入特征提取网络，得到所述辅助图片集对应的第一特征图和所述检测图片集对应的第二特征图；

处理模块，用于使用区域建议网络对所述第一特征图和所述第二特征图进行处理，得到所述检测图片集中预测的缺陷部位；

确定模块，用于确定所述辅助图片集中标注的缺陷部位与所述检测图片集中预测的缺陷部位的相似度；

输出模块，用于根据所述相似度，输出检测结果。

9.一种电子设备，其特征在于，所述电子设备包括处理器、存储器、通信接口以及一个或多个程序，其中，所述一个或多个程序被存储在所述存储器中，并且被配置由所述处理器执行，所述程序包括用于执行权利要求1至7任一项方法中的步骤的指令。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有计算机程序，所述计算机程序被处理器执行以实现权利要求1至7任意一项所述的方法。