[发明专利]电力稀缺缺陷图像生成及检测方法、装置及相关设备

权利要求书

1.一种电力稀缺缺陷图像生成及检测方法，其特征在于，所述方法包括：

获取第一缺陷绝缘子图像以及完整绝缘子图像，并根据所述第一缺陷绝缘子图像将所述完整绝缘子图像转换为缺陷绝缘子图像，得到样本缺陷绝缘子图像；

将所述完整绝缘子图像以及所述样本缺陷绝缘子图像训练缺陷绝缘子检测模型，得到训练好的缺陷绝缘子检测模型；

将待检测图像输入到所述训练好的缺陷绝缘子检测模型进行检测。

2.如权利要求1所述的电力稀缺缺陷图像生成及检测方法，其特征在于，所述根据所述第一缺陷绝缘子图像将所述完整绝缘子图像转换为缺陷绝缘子图像，得到样本缺陷绝缘子图像，包括：

对所述第一缺陷绝缘子图像进行扩充，得到第二缺陷绝缘子图像；

根据所述第二缺陷绝缘子图像将所述完整绝缘子图像转换为缺陷绝缘子图像，得到第三缺陷绝缘子图像；

基于所述第三缺陷绝缘子图像，得到样本缺陷绝缘子图像。

3.如权利要求2所述的电力稀缺缺陷图像生成及检测方法，其特征在于，所述对所述第一缺陷绝缘子图像进行扩充，得到第二缺陷绝缘子图像，包括：

通过数据增强方式对所述第一缺陷绝缘子图像进行扩充，得到第二缺陷绝缘子图像。

4.如权利要求2所述的电力稀缺缺陷图像生成及检测方法，其特征在于，所述根据所述第二缺陷绝缘子图像将所述完整绝缘子图像转换为缺陷绝缘子图像，得到第三缺陷绝缘子图像，包括：

根据所述第二缺陷绝缘子图像，划分第一样本集，根据所述完整绝缘子图像划分第二样本集；

根据所述第一样本集与所述第二样本集训练缺陷绝缘子图像生成网络，得到训练好的缺陷绝缘子图像生成网络；

将所述完整绝缘子图像输入到所述训练好的缺陷绝缘子图像生成网络，得到第三缺陷绝缘子图像。

5.如权利要求4所述的电力稀缺缺陷图像生成及检测方法，其特征在于，所述根据所述第一样本集与所述第二样本集训练缺陷绝缘子图像生成网络，得到训练好的缺陷绝缘子图像生成网络，包括：

构建生成式对抗网络，所述生成式对抗网络包括第一生成网络、第二生成网络、第一判别网络以及第二判别网络；对所述生成式对抗网络的训练包括循环的第一阶段与第二阶段；

其中，所述第一阶段包括：

通过所述第一生成网络，将第二样本集中的完整绝缘子图像转换为第一生成缺陷绝缘子图像；

通过第一判别网络对所述第一生成缺陷绝缘子图像与第一样本集中的缺陷绝缘子图像进行判别，并通过第一判别网络中的第一损失函数对所述第一判别网络进行参数调整；

通过所述第二生成网络，将所述第一生成缺陷绝缘子图像转换为第一生成完整绝缘子图像；

通过第一损失函数，计算第一生成完整绝缘子图像与所述第二样本集中的完整绝缘子图像之间的第二损失，并根据所述第一损失调整所述第一生成网络、第二生成网络的参数；

所述第二阶段包括：

通过所述第二生成网络，将第一样本集中的缺陷绝缘子图像转换为第二生成完整绝缘子图像；

通过第二判别网络对所述第二生成完整绝缘子图像与第二样本集中的完整绝缘子图像进行判别，并通过第二判别网络中的第三损失函数对所述第二判别网络进行参数调整；

通过所述第一生成网络，将所述第二生成完整绝缘子图像转换为第二生成缺陷绝缘子图像；

通过第四损失函数，计算第二生成缺陷绝缘子图像与所述第一样本集中的缺陷绝缘子图像之间的第四损失，并根据所述第四损失调整所述第一生成网络、第二生成网络的参数；

循环迭代所述第一阶段与所述第二阶段，完成训练后，取所述第一生成网络作为目标生成网络。