[发明专利]双树复小波多层信息融合驱动的钢管表面缺陷检测方法

权利要求书

1.双树复小波多层信息融合驱动的钢管表面缺陷检测方法，其特征在于：包括以下步骤：

S1：获取无缝钢管表面缺陷的样本数据库；

S2：建立无缝钢管表面缺陷的智能检测网络；

S3：利用步骤S1获得的样本数据库结合双树复小波对智能检测网络进行训练，得到表面缺陷检测网络模型；

S4：对于任一获取到的无缝钢管表面缺陷待测图像，将其输入至步骤3获得的表面缺陷检测网络模型，若待测图像存在缺陷，则表面缺陷检测网络模型在待测图像对应的缺陷位置进行标记。

2.根据权利要求1所述的双树复小波多层信息融合驱动的钢管表面缺陷检测方法，其特征在于：在步骤S1中，样本数据库的建立方法为：

S11:将被测无缝钢管固定在图像拍摄平台并获取用于训练无缝钢管表面缺陷的表面图像；

S12:将表面图像的RGB转换为YUV空间；

S13:将YUV空间中的Y通道置零并再次转换至RGB图像；

S14:将步骤13中的RGB图像再转换为灰度图像；

S15:在灰度图像中逐像素标注表面图像中缺陷的位置，得到样本缺陷数据库。

3.根据权利要求2所述的双树复小波多层信息融合驱动的钢管表面缺陷检测方法，其特征在于：在步骤S12中，YUV空间的转换方法如下：

其中R、G、B分别为表面图像RGB三通道的信息，Y、U、V分别为转换到YUV空间的通道信息。

4.根据权利要求2所述的双树复小波多层信息融合驱动的钢管表面缺陷检测方法，其特征在于：在步骤S13中，RGB图像的转换方法为：

其中R、G、B分别为转换后RGB图像三通道的信息，U、V分别为YUV空间的两个通道信息。

5.根据权利要求2所述的双树复小波多层信息融合驱动的钢管表面缺陷检测方法，其特征在于：在步骤S2中，所述智能检测网络中设有由CBAM模块和U-Net网络组成的CBAM-U-Net块；所述CBAM模块具有通道注意力机制和空间注意力机制，CBAM模块通过通道注意力机制和空间注意力机制将输入特征进行强化，强化后的特征传递至U-Net网络以进行本层特征状态的提取。

6.根据权利要求5所述的双树复小波多层信息融合驱动的钢管表面缺陷检测方法，其特征在于：所述表面缺陷检测网络模型建立方法为：

S21:对输入的灰度图像进行双树复小波分解，得到双树复小波各层成分；

S22:将双树复小波各层成分分别传递至CBAM-U-Net块；

S23:将各层CBAM-U-Net结果相加并形成无缝钢管表面缺陷神经网络，以无缝钢管表面缺陷神经网络作为智能检测网络。