肖像风格迁移模型训练方法、肖像风格迁移方法以及装置

摘要

本发明公开了一种肖像风格迁移模型训练方法、肖像风格迁移方法以及装置。其中，模型训练方法包括：获取用户肖像图和绘画风格肖像图，一方面由用户肖像图和肖像风格迁移模型生成中间生成图；另一方面基于用户肖像图对绘画风格肖像图及其脸部掩码图进行配准变形；再综合用户肖像图、中间生成图、风格配准图和掩码配准图，计算作为训练目标的总损失对模型进行优化，重复上述过程直至训练完毕。本发明还相应提供了利用上述训练方法得到的模型进行肖像风格迁移处理的软硬件方案。本发明可以训练出补足了现有模型“考虑不周”之处的最佳肖像风格迁移模型，并由此优化后的模型可以生成大大减弱肖像瑕疵的绘画风格用户肖像图，进而改善用户使用体验。