一种图像生成模型的训练方法及相关装置
摘要:
本申请涉及人工智能技术领域,尤其涉及一种图像生成模型的训练方法及相关装置,该方法为:首先,获取包含目标对象的基础图像,并对基础图像进行图像变换,获得训练数据集;每执行设定次数的迭代训练,输出一次候选模型;针对各图像模板,分别执行以下操作:将一个图像模板分别输入至M个候选模型中,获得目标图像,并分别基于M个目标图像各自与一个图像模板的面部相似度,计算M个候选模型各自的投票票数;最后,基于各票数总和,分别获得相应候选模型的模型权重,并对M个候选模型进行融合,获得目标模型。这样使用单张用户照片,自动训练出图像生成模型,无需人工干预和控制,提升了模型的图像生成效果,同时降低了模型的训练成本。
公开/授权文献
0/0