图像生成模型的训练方法及风格化图像生成方法
摘要:
本申请提供了一种图像生成模型的训练方法及风格化图像生成方法,该训练方法采用对比学习的方式训练图像生成模型。在每轮次训练过程中,计算机设备可根据图像生成模型输出的样本风格化图像、对比学习模型输出的各张正例风格化图像和各张负例风格化图像,计算当前训练轮次对应的训练损失值,并据此训练图像生成模型的参数权重。并且,在每轮次训练过程中,计算机设备可根据样本风格化图像、各张正例风格化图像、各张负例风格化图像、图像生成模型在当前训练轮次对应的调整前后的参数权重,动态自适应地更新对比学习网络的参数权重,从而可根据样本学习的难易程度自动调整模型的更新速度,进而能有效加速图像生成模型的收敛,缩减训练耗时。
0/0