[发明专利]一种管道内壁缺陷检测方法及系统

权利要求书

1.一种管道内壁缺陷检测方法，其特征在于，包括以下步骤：

获取管道内壁图像；

根据所述管道内壁图像，确定所述管道内壁图像对应的多个缺陷类别特征图和多个缺陷位置特征图；

根据各个所述缺陷类别特征图，确定至少一个目标类别特征图，以及根据各个所述缺陷位置特征图，确定至少一个目标位置特征图，每个所述目标类别特征图为各所述缺陷类别特征图中满足第一分辨率要求的缺陷类别特征图，每个所述目标位置特征图为各所述缺陷位置特征图中满足第二分辨率要求的缺陷位置特征图；

根据各个所述目标位置特征图，确定每个所述目标位置特征图对应的回归框，对于每个所述回归框，所述回归框对应的区域为该回归框对应的目标位置特征图在所述管道内壁图像上所对应的缺陷所在的区域；

根据各个所述回归框和各个所述目标类别特征图，确定每个所述目标类别特征图对应的分类信息，以及每个所述回归框对应的回归框特征图，所述分类信息为所述目标类别特征图中缺陷的类别，对于每个所述回归框，所述回归框特征图为所述回归框对应的缺陷所在区域的特征图；

根据各个所述分类信息，确定所述管道内壁图像对应的类别信息，所述类别信息为所述管道内壁图像中缺陷的类别；

根据各个所述回归框特征图，确定每个所述回归框特征图对应的回归框分数，所述回归框分数表征了回归框的准确性；

根据各个所述回归框对应的回归框分数和各个所述分类信息，确定所述管道内壁图像对应的定位信息，所述定位信息为所述管道内壁图像中缺陷的定位。

2.根据权利要求1所述的方法，其特征在于，所述根据所述管道内壁图像，确定所述管道内壁图像对应的多个缺陷类别特征图和多个缺陷位置特征图，包括：

将所述管道内壁图像输入特征提取器，通过所述特征提取器的每一层，提取所述管道内壁图像的缺陷类别特征，得到所述缺陷类别特征图，以及提取所述管道内壁图像的缺陷位置特征，得到所述缺陷位置特征图，所述特征提取器中的每一层对应的分辨率不同。

3.根据权利要求2所述的方法，其特征在于，所述根据各个所述缺陷类别特征图，确定至少一个目标类别特征图，以及根据各个所述缺陷位置特征图，确定至少一个目标位置特征图，包括：

S11，获取通过所述特征提取器中的至少两个特征提取层中每个所述特征提取层对应的所述缺陷类别特征图和所述缺陷位置特征图，所述特征提取器中的各所述特征提取层对应的分辨率不同；

S12，从所述至少两个特征提取层中的第一个特征提取层开始直到最后一个特征提取层，将所述第一个特征提取层作为当前特征提取层，重复执行步骤S13，直到所述最后一个特征提取层作为所述当前特征提取层，得到每个所述特征提取层对应的第二类别特征图和第二位置特征图；

S13，将所述当前特征提取层对应的缺陷类别特征图和所述缺陷位置特征图进行下采样，得到第一类别特征图和第一位置特征图，根据所述第一类别特征图和所述第一个特征提取层对应的所述缺陷类别特征图，得到第二类别特征图，根据所述第一位置特征图和所述第一个特征提取层对应的所述缺陷位置特征图，得到第二位置特征图，将所述当前特征提取层的下一层作为新的当前特征提取层；

S14，将每个所述特征提取层对应的第二类别特征图作为所述至少一个目标类别特征图，将每个所述特征提取层对应的第二位置特征图作为所述至少一个目标位置特征图。