用于训练服饰生成模型的方法和用于生成服饰的方法
摘要:
本公开的实施例提供一种用于训练服饰生成模型的方法和一种用于生成服饰的方法。在用于训练服饰生成模型的方法中,获取服饰单品的训练图像集。训练图像集包括多个训练图像组。针对包括第一和第二训练图像的训练图像组:提取第一和第二训练图像的颜色、纹理及形状特征;由服饰生成器来生成颜色、纹理及形状校验图像;根据颜色和纹理校验图像与第一训练图像计算颜色和纹理损失函数;根据形状校验图像与第二训练图像计算重建损失函数;通过真假判别器来判断颜色、纹理和形状校验图像的真实性;通过最小化服饰生成模型的总损失函数来更新服饰生成模型的参数。总损失函数包括:颜色损失函数、纹理损失函数、重建损失函数和真假判别器的损失函数。
0/0