摘要
本申请涉及一种缺陷图像生成方法、装置、计算机设备及存储介质。方法包括:对获取的待生成工业外观缺陷图像进行区域提取,得到外观缺陷区域图像和工业背景区域图像;将外观缺陷区域图像和其对应的缺陷叠加掩膜图像输入至图像生成模型中的编码网络进行处理,输出外观编码特征图和掩膜编码特征图;根据外观编码特征图、掩膜编码特征图和外观缺陷区域图像对应的外观缺陷掩膜图像,确定扩散特征图;将扩散特征图输入至图像生成模型中的解码网络输出目标缺陷区域图像;将外观缺陷区域图像对应的目标缺陷区域图像与工业背景区域图像进行图像融合,得到目标工业外观缺陷图像。采用本申请,能够实现提高工业外观缺陷图像生成的效率和精准度。
技术关键词
编码特征
图像生成模型
掩膜
样本
缺陷类别
解码网络
缺陷轮廓
图像生成方法
生成工业
计算机设备
图像生成装置
索引
文本
模块
图像分割
噪声