一种基于参考图像引导的工业品缺陷样本可控生成方法

本发明涉及一种基于参考图像引导的工业品缺陷样本可控生成方法，通过参考缺陷处理模块和参考缺陷特征编码模块提取参考缺陷的样式信息，并作为去噪UNet的引导信息；使用缺陷位置及细节处理模块和缺陷位置及细节编码模块生成去噪UNet的另一部分引导信息；构建一种基于Gram矩阵的缺陷样式损失函数并训练模型；在推理阶段，根据用户输入的缺陷强度参数，使用训练后的模型，从随机噪声中通过逐步去噪，在给定区域生成指定强度的与参考缺陷类似的图像，并在背景区域保持与输入的背景图像一致。本发明使用缺陷图像引导，根据输入的缺陷强度控制参数，生成多样化的缺陷样本，从而为缺陷检测模型训练提供更加丰富和贴近真实缺陷图像的数据样本。