一种工件缺陷图像生成方法、装置、设备及介质

本申请的实施例提供了一种工件缺陷图像生成方法、装置、设备及介质，首先对输入的高分辨率缺陷图像裁剪得到包含生成缺陷区域的切片，对其裁剪得到包含缺陷区域的图片，之后进行数据增强操作之后依次输入至CLIP图片编码器、多层感知器得到特征向量，并对其进行掩码形状增强处理生成带有缺陷掩码的图像，将带有缺陷掩码的图像与高斯噪声拼接之后与特征向量一同输入至扩散模型进行训练之后扩散得到生成图片，将生成图片粘贴至输入的高分辨率缺陷图像之后输入变分自编码器进行解码得到原分辨率缺陷图片。加入lora使得该方法更适配缺陷领域的生成，将低维特征向量恢复为生成高质量、高分辨率的缺陷图像，确保生成的图像具有较高的清晰度和细节。