电力设备缺陷图像扩散生成方法及系统

本发明公开了一种电力设备缺陷图像扩散生成方法及系统，涉及电力视觉技术领域，包括步骤：将正常原始图像划分为多个子区域图像并进行位置编码；进行降维和映射得到低维空间的原始特征图和文本特征嵌入：添加高斯噪声得到噪声特征图；进行特征提取得到第一特征图和第二特征图；得到融合特征图；去除缺陷噪声得到映射特征图；若满足预设终止条件，则采用解码器将映射特征图还原到三维空间，得到与拟定缺陷文本提示对应的电力设备缺陷扩散图像，若不满足预设终止条件，则将映射特征图更新为噪声特征图并重复上述步骤。本发明的方法，兼顾了生成质量和结果的多样性与真实性，能够为下游视觉任务提供全新的训练样本，同时所需算力资源较少、易于实施。