[发明专利]一种识别不同类型焊口的方法

权利要求书

1.一种识别不同类型焊口的方法，其特征在于，包括以下步骤：

S1、采集X射线焊口图像，构建标注数据集；

S2、采用标注数据集对焊口定位识别模型进行训练，得到训练完成的焊口定位识别模型；

所述步骤S2中焊口定位识别模型包括：第一深度可分离卷积层、第二深度可分离卷积层、第三深度可分离卷积层、第四深度可分离卷积层、第五深度可分离卷积层、第六深度可分离卷积层、第七深度可分离卷积层、第八深度可分离卷积层、第九深度可分离卷积层、第十深度可分离卷积层、第十一深度可分离卷积层、第一最大池化层、第二最大池化层、第三最大池化层、第四最大池化层、第五最大池化层、第六最大池化层、第七最大池化层、第八最大池化层、填零式最大池化层、第一拼接层、第二拼接层、上采样层和检测器；

所述第一深度可分离卷积层的输入端作为焊口定位识别模型的输入端，其输出端与第一最大池化层的输入端连接；所述第一最大池化层的输出端与第二深度可分离卷积层的输入端连接；所述第二深度可分离卷积层的输出端与第二最大池化层的输入端连接；所述第二最大池化层的输出端与第三深度可分离卷积层的输入端连接；所述第三深度可分离卷积层的输出端与第三最大池化层的输入端连接；所述第三最大池化层的输出端与第四深度可分离卷积层的输入端连接；所述第四深度可分离卷积层的输出端与第四最大池化层的输入端连接；所述第四最大池化层的输出端与第五深度可分离卷积层的输入端连接；所述第五深度可分离卷积层的输出端与第五最大池化层的输入端连接；所述第五最大池化层的输出端与第六深度可分离卷积层的输入端连接；所述第六深度可分离卷积层的输出端分别与填零式最大池化层的输入端和第二拼接层的第一输入端连接；所述填零式最大池化层的输出端分别与第一拼接层的输入端、第六最大池化层的输入端、第七最大池化层的输入端、第八最大池化层的输入端、第六最大池化层的输出端、第七最大池化层的输出端和第八最大池化层的输出端连接；所述第一拼接层的输出端与第七深度可分离卷积层的输入端连接；所述第七深度可分离卷积层的输出端与第九深度可分离卷积层的输入端连接；所述第九深度可分离卷积层的输出端分别与第八深度可分离卷积层的输入端和第十深度可分离卷积层的输入端连接；所述第八深度可分离卷积层的输出端与上采样层的输入端连接；所述上采样层的输出端与第二拼接层的第二输入端连接；所述第二拼接层的输出端与第十一深度可分离卷积层的输入端连接；所述第十一深度可分离卷积层的输出端与检测器的第一输入端连接；所述第十深度可分离卷积层的输出端与检测器的第二输入端连接；所述检测器的输出端作为焊口定位识别模型的输出端；

S3、采用训练完成的焊口定位识别模型对需进行焊口识别的图像进行识别，得到焊口的位置和类型。

2.根据权利要求1所述的识别不同类型焊口的方法，其特征在于，所述步骤S1包括以下分步骤：

S11、采集X射线焊口图像，得到焊口图像数据集；

S12、对焊口图像数据集进行分类，得到不同类型的图像数据集；

S13、对不同类型的图像数据集进行图像增强处理，得到增强图像数据集；

S14、对增强图像数据集中非正常的焊口图像进行标注，得到标注数据集。

3.根据权利要求2所述的识别不同类型焊口的方法，其特征在于，所述步骤S13中图像增强处理包括：提高图像的对比度、提高图像的亮度、水平翻转、垂直翻转、顺时针旋转90度和逆时针旋转90度。

4.根据权利要求2所述的识别不同类型焊口的方法，其特征在于，所述步骤S14中标注包括：对焊口位置的标注和焊口类型的标注。