虚拟图像生成模型的训练方法、装置、设备及介质

文档序号:33124312发布日期:2023-02-01 04:39阅读:来源:国知局

技术特征:
1.一种虚拟图像生成模型的训练方法,其特征在于,包括:从标准模板图库中选定需要生成虚拟图像的标准模板图,从所述标准模板图对应的数据集中抽取真实图像;基于所述真实图像,生成所述真实图像对应的第一噪声图;其中,所述第一噪声图用于表征所述真实图像与所述标准模板图之间的差异;合并所述标准模板图和所述第一噪声图,得到合并图像;将所述合并图像输入至预先设置的生成器网络中,得到所述生成器网络输出的候选图像;其中,所述生成器网络用于基于所述合并图像生成虚拟的图像作为候选图像;将第一图像对和第二图像对输入至预先设置的判别器网络,以由所述判别器网络对抗训练所述生成器网络,得到训练后的生成器网络,作为虚拟图像生成模型;其中,所述第一图像对包括所述合并图像和所述真实图像,所述第二图像对包括所述合并图像和所述候选图像。2.根据权利要求1所述的虚拟图像生成模型的训练方法,其特征在于,在所述将第一图像对和第二图像对输入至预先设置的判别器网络,以由所述判别器网络对抗训练所述生成器网络,得到训练后的生成器网络,作为虚拟图像生成模型之后,所述方法还包括:将所述标准模板图和随机生成的第二噪声图合并后,输入至所述虚拟图像生成模型中,得到所述虚拟图像生成模型输出的所述标准模板图对应的虚拟图像。3.根据权利要求2所述的虚拟图像生成模型的训练方法,其特征在于,在所述将所述标准模板图和随机生成的第二噪声图合并后,输入至所述虚拟图像生成模型中,得到所述虚拟图像生成模型输出的所述标准模板图对应的虚拟图像之后,所述方法还包括:基于所述虚拟图像生成模型输出的所述标准模板图对应的虚拟图像,训练图像感知模型。4.根据权利要求1至3任一项所述的虚拟图像生成模型的训练方法,其特征在于,所述从标准模板图库中选定需要生成虚拟图像的标准模板图,从所述标准模板图对应的数据集中抽取真实图像,包括:从标准模板图库中选定所述标准模板图;从所述标准模板图对应的数据集中,通过随机抽样或非随机抽样抽取所述真实图像。5.根据权利要求1至3任一项所述的虚拟图像生成模型的训练方法,其特征在于,所述基于所述真实图像,生成所述真实图像对应的第一噪声图,包括:提取所述真实图像中的图像特征,作为噪声种子;基于所述噪声种子,生成所述真实图像对应的第一噪声图;其中,所述第一噪声图的尺寸与所述标准模板图的尺寸相同。6.根据权利要求1至3任一项所述的虚拟图像生成模型的训练方法,其特征在于,所述合并所述标准模板图和所述第一噪声图,得到合并图像,包括:通过通道级的图像拼接和/或矩阵计算,合并所述标准模板图和所述第一噪声图,得到所述合并图像。7.根据权利要求1至3任一项所述的虚拟图像生成模型的训练方法,其特征在于,在所述将第一图像对和第二图像对输入至预先设置的判别器网络,以由所述判别器网络对抗训练所述生成器网络,得到训练后的生成器网络,作为虚拟图像生成模型之前,所述方法还包
括:将所述合并图像和所述真实图像作为所述第一图像对,设定所述第一图像对的标签值表征为真;将所述合并图像和所述候选图像作为所述第二图像对,设定所述第二图像对的标签值表征为假;所述将第一图像对和第二图像对输入至预先设置的判别器网络,以由所述判别器网络对抗训练所述生成器网络,得到训练后的生成器网络,作为虚拟图像生成模型,包括:将所述第一图像对和所述第二图像对输入至所述判别器网络,以由所述判别器网络基于所述第一图像对的标签值和所述第二图像对的标签值,对抗训练所述生成器网络,得到训练后的生成器网络,作为虚拟图像生成模型。8.一种虚拟图像生成模型的训练装置,其特征在于,包括:抽取模块,用于从标准模板图库中选定需要生成虚拟图像的标准模板图,从所述标准模板图对应的数据集中抽取真实图像;生成模块,用于基于所述真实图像,生成所述真实图像对应的第一噪声图;其中,所述第一噪声图用于表征所述真实图像与所述标准模板图之间的差异;合并模块,用于合并所述标准模板图和所述第一噪声图,得到合并图像;处理模块,用于将所述合并图像输入至预先设置的生成器网络中,得到所述生成器网络输出的候选图像;其中,所述生成器网络用于基于所述合并图像生成虚拟的图像作为候选图像;训练模块,用于将第一图像对和第二图像对输入至预先设置的判别器网络,以由所述判别器网络对抗训练所述生成器网络,得到训练后的生成器网络,作为虚拟图像生成模型;其中,所述第一图像对包括所述合并图像和所述真实图像,所述第二图像对包括所述合并图像和所述候选图像。9.一种电子设备,包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序,其特征在于,所述处理器执行所述程序时实现如权利要求1至7任一项所述虚拟图像生成模型的训练方法。10.一种非暂态计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至7任一项所述虚拟图像生成模型的训练方法。11.一种计算机程序产品,包括计算机程序,其特征在于,所述计算机程序被处理器执行时实现如权利要求1至7任一项所述虚拟图像生成模型的训练方法。

技术总结
本发明提供一种虚拟图像生成模型的训练方法、装置、设备及介质,方法包括:从标准模板图库中选定需要生成虚拟图像的标准模板图,从标准模板图对应的数据集中抽取真实图像;基于真实图像,生成真实图像对应的第一噪声图;合并标准模板图和第一噪声图,得到合并图像;将合并图像输入至预先设置的生成器网络中,得到生成器网络输出的候选图像;将第一图像对和第二图像对输入至预先设置的判别器网络,以由判别器网络对抗训练生成器网络,得到训练后的生成器网络,作为虚拟图像生成模型;其中,第一图像对包括合并图像和真实图像,第二图像对包括合并图像和候选图像。通过生成对抗网络训练得到虚拟图像生成模型,提高了其生成虚拟图像的真实性。真实性。真实性。


技术研发人员:王建功 王飞跃 王晓 王雨桐 田永林
受保护的技术使用者:中国科学院自动化研究所
技术研发日:2022.09.20
技术公布日:2023/1/31
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1