[发明专利]用于分辨排水管道缺陷类型及自动鉴别缺陷等级的方法

权利要求书

1.用于分辨排水管道缺陷类型及自动鉴别缺陷等级的方法，其特征在于，包括如下步骤：

S1、采用实际的排水管道缺陷图像构建数据集，并且数据集中所有的图像均添加有缺陷类型和缺陷等级的数据标签；

S2、将数据集进行切分，划分为训练集和测试集，将训练集送入未训练的主分类器模型中；

S3、主分类器模型采用改进的VGG－16卷积神经网络架构，所述改进的VGG－16卷积神经网络架构通过在原有VGG－16卷积神经网络架构的全连接层后依次加入256个神经元的fc1全连接层和11个神经元的fc2全连接层后得到改进的VGG－16卷积神经网络架构；

经过训练集对主分类器模型进行参数训练，然后用测试集进行检测，最终得到训练好的主分类器模型，保存待用；

S4、将训练集按照缺陷类型进行分组，对每组同类缺陷图像分组为次分类器训练集和次分类器测试集，并且对每组同类缺陷图像建立一个新的次分类器；按照主分类器模型的架构，通过将主分类器模型中的11个神经元的fc2全连接层改为5个神经元的fc2全连接层得到次分类器；次分类器各层参数权重与主分类器相同，fc2层至softmax层全部初始化为期望为0的高斯随机分布；

S5、对于每个次分类器，固定其前10层，超参数设置和主分类器相同；每组的次分类器用该组的次分类器训练集进行训练后用该组的次分类器测试集检验，得到次分类器模型；

S6、将采集得到的待分辨的排水管道缺陷图像输入主分类器，识别得到图像中排水管道的缺陷类型，再根据缺陷类型输入对应的次分类器，识别得到图像中排水管道的缺陷等级。

2.根据权利要求1所述的方法，其特征在于，进行步骤S2之前，先对步骤S1构建得到的数据集进行增强处理，通过翻转和裁切操作增大数据集的数据量，并对数据集中的所有图像添加高斯噪声和椒盐噪声。

3.根据权利要求2所述的方法，其特征在于，进行步骤S2之前，增强处理后，对数据集中每一张图像进行归一化处理。

4.根据权利要求1所述的方法，其特征在于，步骤S2中，以数量比8：2的比例将数据集划分为训练集和测试集。

5.根据权利要求1所述的方法，其特征在于，步骤S3中，主分类器模型的超参数分别初始化设置为：学习率learning＿rate＝0.001，批大小batch＿size＝10，训练代数epoch＝50；每十代将学习率缩小一半，即new＿learning＿rate＝learning＿rate/2。

6.根据权利要求1所述的方法，其特征在于，步骤S6中，将训练集按数量比8：2划分为次分类器训练集和次分类器测试集。