【技术实现步骤摘要】
本申请涉及图像处理，特别是涉及一种缺陷图像生成方法、装置、计算机设备及计算机可读存储介质。
技术介绍
1、随着人工智能技术的发展，采用视觉非接触式的自动化工业检测，以替代繁琐的人工检测，而针对自动化工业检测所对应的深度神经网络算法，通常需要大量的缺陷图像进行训练。
2、为解决训练所需的缺陷图像难以收集的问题，一方面，采用现有的对抗生成网络生成缺陷图像，然而该方法易于出现模式坍塌、生成质量低等问题；另一方面，采用现有的扩散模型生成缺陷图像，然而扩散模型的生成过程过于简化，缺乏对应的引导条件，无法有效地捕捉缺陷的复杂性和多样性，从而难以精准地控制所需生成缺陷的缺陷类型和缺陷区域，再者，扩散模型通常是针对一个特定的缺陷进行训练的，从而无法同时生成多类缺陷。
技术实现思路
1、基于此，有必要针对上述技术问题，提供一种缺陷图像生成方法、装置、计算机设备、计算机可读存储介质及计算机程序产品，能够实现高效地、准确地生成在指定区域携带有指定类别的缺陷的缺陷图像。
2、第一方面，本申请提供了一种缺陷图像生成方法，包括：
3、获取原始图像和其对应的待生成缺陷信息，待生成缺陷信息包括缺陷区域信息和缺陷类别信息；
4、基于原始图像、待生成缺陷信息和已训练的缺陷图像生成模型，确定原始图像对应的生成缺陷图像；
5、基于原始图像和生成缺陷图像，确定目标缺陷图像；目标缺陷图像在缺陷区域信息所对应的缺陷区域中携带有缺陷类别信息所对应的缺陷。
6
7、获取模块，用于获取原始图像和其对应的待生成缺陷信息，待生成缺陷信息包括缺陷区域信息和缺陷类别信息；
8、第一确定模块，用于基于原始图像、待生成缺陷信息和已训练的缺陷图像生成模型，确定原始图像对应的生成缺陷图像；
9、第二确定模块，用于基于原始图像和生成缺陷图像，确定目标缺陷图像；目标缺陷图像在缺陷区域信息所对应的缺陷区域中携带有缺陷类别信息所对应的缺陷。
10、第三方面，本申请提供了一种计算机设备，计算机设备包括存储器和处理器，存储器存储有计算机程序，处理器执行该计算机程序时实现上述的方法中的步骤。
11、第四方面，本申请提供了一种计算机可读存储介质，其上存储有计算机程序，该计算机程序被处理器执行时实现上述的方法中的步骤。
12、第五方面，本申请提供了一种计算机程序产品，计算机程序产品包括计算机程序，该计算机程序被处理器执行时实现上述的方法中的步骤。
13、上述缺陷图像生成方法、装置、计算机设备、计算机可读存储介质及计算机程序产品，获取原始图像和其对应的待生成缺陷信息，其中待生成缺陷信息包括缺陷区域信息和缺陷类别信息，其中缺陷区域信息指示待生成缺陷所对应的缺陷区域，缺陷类别信息指示待生成缺陷所对应的缺陷类别；基于原始图像和待生成缺陷信息，由已训练的缺陷图像生成模型进行对应的图像处理并输出原始图像对应的生成缺陷图像；基于原始图像和生成缺陷图像之间的对应关系，在原始图像的基础上确定包含有生成缺陷图像信息的目标缺陷图像；基于此，高效地、准确地生成在缺陷区域信息所对应的缺陷区域中携带有缺陷类别信息所对应的缺陷的目标缺陷图像。
本文档来自技高网...【技术保护点】
1.一种缺陷图像生成方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述基于所述原始图像、所述待生成缺陷信息和已训练的缺陷图像生成模型，确定所述原始图像对应的生成缺陷图像，包括：
3.根据权利要求2所述的方法，其特征在于，所述将所述目标文本条件编码、所述目标语义条件编码、所述目标特征图输入至已训练的缺陷图像生成模型之前，所述方法还包括：
4.根据权利要求2所述的方法，其特征在于，所述基于所述缺陷区域信息，确定所述原始图像对应的缺陷区域图、背景区域图和二值掩膜图，包括：
5.根据权利要求1或2所述的方法，其特征在于，所述基于所述原始图像和所述生成缺陷图像，确定目标缺陷图像，包括：
6.根据权利要求2所述的方法，其特征在于，所述将所述缺陷区域图、所述背景区域图和所述二值掩膜图进行融合处理，得到目标特征图，包括：
7.根据权利要求2所述的方法，其特征在于，所述针对所述缺陷类别信息，生成对应的文本描述信息，包括：
8.一种缺陷图像生成装置，其特征在于，包括：
9.一种计算机
10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现权利要求1至7中任一项所述的方法的步骤。
...【技术特征摘要】
1.一种缺陷图像生成方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述基于所述原始图像、所述待生成缺陷信息和已训练的缺陷图像生成模型，确定所述原始图像对应的生成缺陷图像，包括：
3.根据权利要求2所述的方法，其特征在于，所述将所述目标文本条件编码、所述目标语义条件编码、所述目标特征图输入至已训练的缺陷图像生成模型之前，所述方法还包括：
4.根据权利要求2所述的方法，其特征在于，所述基于所述缺陷区域信息，确定所述原始图像对应的缺陷区域图、背景区域图和二值掩膜图，包括：
5.根据权利要求1或2所述的方法，其特征在于，所述基于所述原始图像和所述生成缺陷图像，确定目标缺陷图像，包括...
【专利技术属性】
技术研发人员：巫文良，陈鹏光，倪铭昊，王朝云，姚恒志，刘枢，吕江波，沈小勇，
申请(专利权)人：深圳思谋信息科技有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。