【技术实现步骤摘要】
本公开属于计算机,具体涉及一种图像生成方法及装置、电子设备、存储介质。
技术介绍
1、随着计算机技术的发展,扩散模型(diffusion model,dm)等图像生成模型应运而生。在工业产品的缺陷检测中,可以利用扩散模型来生成带有缺陷的样本图像,以提高工业产品的缺陷检测的准确率。
2、然而目前生成的带有缺陷的样本图像,并没有准确的标签来标注出缺陷的类型,需要人工进行标注,极大的浪费时间,不利于提高生产效率。
技术实现思路
1、本公开旨在至少解决现有技术中存在的技术问题之一,提供了一种图像生成方法及装置、电子设备、存储介质。
2、第一方面,本公开实施例提供了一种图像生成方法,所述图像生成方法包括:
3、利用第一生成模型,对类别标签进行处理,生成掩膜标签;
4、利用提取模型,对目标图像进行处理,提取所述目标图像的目标特征;
5、利用第二生成模型,对所述掩膜标签、所述目标特征和原始样本图像进行处理,生成带有目标掩膜的样本图像。
< ...【技术保护点】
1.一种图像生成方法,其特征在于,所述图像生成方法包括:
2.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,之后还包括:
3.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,包括:
4.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述类别和所述掩膜标签,对所述第一生成模型进行训练,包括:
5.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述第一生成模型,根据所述类别标签,预测生成所述掩膜标签,
...
【技术特征摘要】
1.一种图像生成方法,其特征在于,所述图像生成方法包括:
2.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,之后还包括:
3.根据权利要求1所述的图像生成方法,其特征在于,所述利用第一生成模型,对类别标签进行处理,生成掩膜标签,包括:
4.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述类别和所述掩膜标签,对所述第一生成模型进行训练,包括:
5.根据权利要求3所述的图像生成方法,其特征在于,所述利用所述第一生成模型,根据所述类别标签,预测生成所述掩膜...
【专利技术属性】
技术研发人员:郑瑞,
申请(专利权)人:京东方科技集团股份有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。