摘要
本发明实施例公开了一种基于扩散模型的训练样本生成方法、装置以及计算机设备,应用于文本图像生成中,其中,所述方法包括:基于现有的训练样本集选择类别不同的第一样本对,然后基于预设的匹配模型和扩散模型计算对应的标签预测损失值,并基于该损失值对匹配模型进行模型更新;然后基于扩散模型生成与第一样本对对应的第二样本对,并基于匹配模型进行特征计算然后计算第一样本对和第二样本对之间的差异损失值,然后基于该损失值进行匹配模型和扩散模型的更新,然后对于待生成样本,基于完成模型更新的扩散模型生成新的生成样本。采用本发明实施例,可以提高缺陷样本生成的质量和效率。
技术关键词
训练样本生成方法
模型更新
预测类别
标签
计算机设备
扩充模块
训练样本集
文本
生成装置
图像
存储器
处理器
系统为您推荐了相关专利信息
三维场景重建方法
三维模型
特征匹配算法
低成本
数据
土壤体积含水量
动态预测方法
饱和导水率
积水
土壤入渗能力