图像生成方法及装置、电子设备、存储介质制造方法及图纸

技术编号:44966109 阅读:23 留言:0更新日期:2025-04-12 01:38
本公开提供一种图像生成方法及装置、电子设备、存储介质,属于计算机技术领域,其可解决现有的带有缺陷的样本图像没有准确的标签来标注出缺陷的类型,需要人工进行标注,极大的浪费时间,不利于提高生产效率的问题。本公开的图像生成方法包括:利用第一生成模型,对类别标签进行处理,生成掩膜标签;利用提取模型,对目标图像进行处理,提取目标图像的目标特征;利用第二生成模型,对掩膜标签、目标特征和原始样本图像进行处理,生成带有目标掩膜的样本图像。

【技术实现步骤摘要】

本公开属于计算机,具体涉及一种图像生成方法及装置、电子设备、存储介质


技术介绍

1、随着计算机技术的发展,扩散模型(diffusion model,dm)等图像生成模型应运而生。在工业产品的缺陷检测中,可以利用扩散模型来生成带有缺陷的样本图像,以提高工业产品的缺陷检测的准确率。

2、然而目前生成的带有缺陷的样本图像,并没有准确的标签来标注出缺陷的类型,需要人工进行标注,极大的浪费时间,不利于提高生产效率。


技术实现思路

1、本公开旨在至少解决现有技术中存在的技术问题之一,提供了一种图像生成方法及装置、电子设备、存储介质。

2、第一方面,本公开实施例提供了一种图像生成方法,所述图像生成方法包括:

3、利用第一生成模型,对类别标签进行处理,生成掩膜标签;

4、利用提取模型,对目标图像进行处理,提取所述目标图像的目标特征;

5、利用第二生成模型,对所述掩膜标签、所述目标特征和原始样本图像进行处理,生成带有目标掩膜的样本图像。

<p>6、在一些实施例本文档来自技高网...

【技术保护点】

1.一种图像生成方法,其特征在于,所述图像生成方法包括:

2.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,之后还包括:

3.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,包括:

4.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述类别和所述掩膜标签,对所述第一生成模型进行训练,包括:

5.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述第一生成模型,根据所述类别标签,预测生成所述掩膜标签,包括:

...

【技术特征摘要】

1.一种图像生成方法,其特征在于,所述图像生成方法包括:

2.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,之后还包括:

3.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,包括:

4.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述类别和所述掩膜标签,对所述第一生成模型进行训练,包括:

5.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述第一生成模型,根据所述类别标签,预测生成所述掩膜...

【专利技术属性】
技术研发人员:郑瑞
申请(专利权)人:京东方科技集团股份有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1