[发明专利]图像生成模型训练方法、图像生成方法、装置及电子设备

权利要求书

1.一种图像生成模型训练方法，其特征在于，包括：

获取相同的第一原始图像生成器和第二原始图像生成器；

获取图像风格不同的第一样本图像和第二样本图像；

生成随机向量，将所述随机向量分别输入所述第一原始图像生成器和所述第二原始图像生成器，得到第一输出图像和第二输出图像；

根据所述第一输出图像、所述第二输出图像、所述第一样本图像和所述第二样本图像训练所述第一原始图像生成器、第一判别器、所述第二原始图像生成器和第二判别器，得到所述第一图像生成器和所述第二图像生成器；所述第一图像生成器生成的第一图像和所述第一样本图像风格相同，所述第二图像生成器生成的第二图像和所述第二样本图像风格相同。

2.根据权利要求1所述的图像生成模型训练方法，其特征在于，所述训练所述第一原始图像生成器、第一判别器、所述第二原始图像生成器和第二判别器，得到所述第一图像生成器和所述第二图像生成器包括：

将所述第一输出图像和所述第一样本图像输入第一判别器，得到所述第一输出图像对应的第一子判别结果和所述第一样本图像对应的第二子判别结果；

将所述第二输出图像和所述第二样本图像输入第二判别器，得到所述第二输出图像对应的第三子判别结果和所述第二样本图像对应的第四子判别结果；

基于所述第一输出图像、所述第二输出图像、所述第一子判别结果、所述第二子判别结果、所述第三子判别结果和所述第四子判别结果训练所述第一原始图像生成器、所述第一判别器、所述第二原始图像生成器和所述第二判别器，得到所述第一图像生成器和所述第二图像生成器。

3.根据权利要求2所述的图像生成模型训练方法，其特征在于，所述基于所述第一输出图像、所述第二输出图像、所述第一子判别结果、所述第二子判别结果、所述第三子判别结果和所述第四子判别结果训练所述第一原始图像生成器、所述第一判别器、所述第二原始图像生成器和所述第二判别器，得到所述第一图像生成器和所述第二图像生成器包括：

根据所述第一子判别结果、所述第二子判别结果、所述第一输出图像的对应于所述第一判别器的第一标注信息和第一样本图像的标注信息确定第一目标损失；

根据所述第三子判别结果、所述第四子判别结果、所述第二输出图像的对应于所述第二判别器的第一标注信息和第二样本图像的标注信息确定第二目标损失；

根据所述第一目标损失训练所述第一判别器，且根据所述第二目标损失训练所述第二判别器；

基于所述第一子判别结果和所述第一输出图像的对应于所述第一原始图像生成器的第二标注信息确定第五判别损失；

基于所述第三子判别结果和所述第二输出图像的对应于所述第二原始图像生成器的第二标注信息确定第六判别损失；

基于所述约束损失函数确定所述第一输出图像和所述第二输出图像的之间的图像内容损失；

根据所述第五判别损失、所述第六判别损失和所述图像内容损失训练所述第一原始图像生成器和所述第二原始图像生成器，

在满足迭代终止条件的情况下，得到所述第一图像生成器和所述第二图像生成器。

4.根据权利要求3所述的图像生成模型训练方法，其特征在于，所述在满足迭代终止条件的情况下，得到所述第一图像生成器和所述第二图像生成器包括：

循环训练所述第一判别器、所述第二判别器、所述第一原始图像生成器和所述第二原始图像生成器，直至满足迭代终止条件；

将训练后的所述第一原始图像生成器确定为所述第一图像生成器，将训练后的所述第二原始图像生成器确定为所述第二图像生成器。