摘要
本发明公开了一种基于条件扩散模型和WGAN的电力图像中毒检测方法,包括以下步骤:步骤1、选取干净图像样本构建初始训练集,训练一个条件扩散模型,使用训练好的条件扩散模型生成真实图像样本数据;步骤2、构建WGAN网络,WGAN网络包括生成器和判别器,其中生成器在输出生成图像样本,判别器获取真实图像样本数据以及生成器输出的生成图像样本进行训练,得到训练好的判别器;步骤3、将待检测电力图像样本输入至步骤2得到的训练的判别器,以判断待检测电力图像是否中毒。本发明形成了一种针对电力图像数据的高效、精确、通用的中毒检测方案。
技术关键词
图像
解码模型
样本
电力
损失函数优化
训练集
编码
数据
网络
随机噪声
标签
参数
误差