一种基于扩散的光伏电池缺陷检测语义数据增强方法

本发明公开了一种基于扩散的光伏电池缺陷检测语义数据增强方法，包括步骤(1)训练光伏电池图像的条件扩散模型和步骤(2)光伏电池图像合成。本发明提出的一种基于扩散的语义数据增强技术，用于光伏电池缺陷检测。语义标签被明确建模为条件信息，以指导语义感知的电池图像合成过程。由于初始输入是通过逐步向真实图像添加高斯噪声而生成的，从而相比于经典扩散图像生成结果图像保留了更为丰富详细的结构信息，因此可以生成更接近真实数据的高质量图像，该发明方法可以有效地增加样本数据多样性，从而提高模型的泛化能力。