一种人物形象相似性模型的训练方法及装置

本发明公开了一种人物形象相似性模型的训练方法及装置，涉及虚拟形象构建技术领域。所述方法包括：生成大量随机的游戏头像图片，同时根据游戏的设置打上对应的第一标签；将游戏头像图片与对应的第一标签基于Stable Diffusion微调一个能生成游戏风格样式的人物头像图片的生成模型；获取真实人脸图片及对应的第二标签，然后输入微调后的生成模型生成游戏头像图片，获得真实人脸图片和生成的游戏头像图片成对数据；使用前述成对数据训练特征提取模型，使用损失函数优化特征的表达；将真实人脸图片或游戏头像图片输入训练完成后的特征提取模型能得到输出的特征向量，用于评估形象间的相似性。本发明解决了既存方法因训练素材少、单一而无法完成训练的问题。