[发明专利]绝缘子图像分割方法、装置、电子设备及存储介质

权利要求书

1.一种绝缘子图像分割方法，其特征在于，包括：

获取包含绝缘子的待分割图像；

将所述待分割图像输入至绝缘子图像分割模型，得到所述绝缘子图像分割模型输出的所述待分割图像的绝缘子分割结果；

其中，所述绝缘子图像分割模型是基于绝缘子图像分割评估模型的生成器确定的；所述绝缘子图像分割评估模型是基于条件生成对抗网络构建的；所述生成器是基于Unet神经网络构建的。

2.根据权利要求1所述的绝缘子图像分割方法，其特征在于，所述绝缘子图像分割模型包括对称设置的编码器部分和解码器部分；

所述编码器部分中的每一非对称卷积层由卷积函数、批量归一化函数和线性整流函数构成；所述解码器部分中的每一非对称卷积层由反卷积函数、批量归一化函数和线性整流函数构成。

3.根据权利要求1所述的绝缘子图像分割方法，其特征在于，所述绝缘子图像分割评估模型的判别器至少包括三个非对称卷积层；

所述判别器中的每一非对称卷积层由卷积函数、批量归一化函数和带泄露线性整流函数构成。

4.根据权利要求1所述的绝缘子图像分割方法，其特征在于，所述绝缘子图像分割模型是基于如下步骤训练得到的：

确定训练样本集；所述训练样本集包括样本绝缘子分割图像、样本绝缘子分割图像对应的绝缘子分割结果；

将所述训练样本集输入至所述绝缘子图像分割评估模型进行训练，若所述绝缘子图像分割评估模型的训练损失小于预设损失阈值或者训练步数达到预设训练步数，则将所述绝缘子图像分割评估模型中的生成器作为所述绝缘子图像分割模型。

5.根据权利要求3所述的绝缘子图像分割方法，其特征在于，所述确定样本绝缘子分割图像，包括：

获取包含绝缘子的原始图像；

基于预设亮度增强系数，对所述原始图像进行亮度增强处理；和/或，基于预设旋转角度，对所述原始图像进行角度旋转处理；

基于亮度增强处理和/或角度旋转处理后的原始图像，确定所述样本绝缘子分割图像。

6.根据权利要求5所述的绝缘子图像分割方法，其特征在于，所述基于亮度增强处理和/或角度旋转处理后的原始图像，确定所述样本绝缘子分割图像，包括：

基于预设图像尺寸，对所述亮度增强处理和/或角度旋转处理后的原始图像进行缩放，并对缩放后的原始图像中的绝缘子部分进行标注，确定所述样本绝缘子分割图像。

7.根据权利要求1至6任一项所述的绝缘子图像分割方法，其特征在于，所述绝缘子图像分割模型是基于如下步骤进行评估的：

确定测试样本集；所述测试样本集与所述训练样本集是基于相同的原始图像确定的；

将所述测试样本集输入至所述绝缘子图像分割评估模型进行测试，确定所述测试样本集对应的均交并比和平均检测时间；

基于所述测试样本集对应的均交并比和平均检测时间，对所述绝缘子图像分割模型进行性能评估。

8.一种绝缘子图像分割装置，其特征在于，包括：

获取单元，用于获取包含绝缘子的待分割图像；

分割单元，用于将所述待分割图像输入至绝缘子图像分割模型，得到所述绝缘子图像分割模型输出的所述待分割图像的绝缘子分割结果；

其中，所述绝缘子图像分割模型是基于绝缘子图像分割评估模型的生成器确定的；所述绝缘子图像分割评估模型是基于条件生成对抗网络构建的；所述生成器是基于Unet神经网络构建的。

9.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至7任一项所述绝缘子图像分割方法的步骤。

10.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至7任一项所述绝缘子图像分割方法的步骤。