服饰生成模型的训练、生成服饰图像的方法和装置

摘要

本公开提供了服饰生成模型的训练、生成服饰图像的方法和装置，涉及人工智能技术领域，具体为增强现实AR、虚拟现实、计算机视觉、深度学习等技术领域，可应用于元宇宙等场景。具体实现方案为：获取服饰图像的样本并训练形状编码器和纹理编码器；分别通过形状编码器和纹理编码器得到的形状特征和纹理特征中分别选择一部分进行拼接，得到拼接特征；将拼接特征输入预训练模型，输出虚拟服饰图像；基于原始图像和虚拟服饰图像之间的差异调整形状编码器、纹理编码器和预训练模型的相关参数；基于调整后的形状编码器、纹理编码器和预训练模型获得服饰生成模型。通过该实施方式能够得到一种可以生成指定款式的服饰图像的模型。