一种工业元件的缺陷图像生成方法

摘要文本

本发明涉及图像处理技术领域，解决了由于缺陷图片样本数据稀缺而无法满足模型训练需求的技术问题，尤其涉及一种工业元件的缺陷图像生成方法，该方法包括以下步骤：S1、基于Inpainting method模型进行预训练获得图像处理模型Unet的预训练权重；S2、采用预训练权重和不同场景下的图像分别对图像处理模型Unet进行训练生成缺陷图像生成模型；S3、将需要生成缺陷的纯背景图像导入至缺陷图像生成模型中完成指定缺陷细节的图像生成。本发明受开源模型Inpainting method的启发，并采用大量开创性设计，使得生成的缺陷的种类、位置以及形状可以被从前端指定或从已有的数据库中提取。