[发明专利]风格图像生成方法、模型训练方法、装置和设备

说明书

本公开实施例涉及一种风格图像生成方法、模型训练方法、装置和设备，该风格图像生成方法包括：获取原始人脸图像；利用预先训练的风格图像生成模型，得到与原始人脸图像对应的目标风格图像；风格图像生成模型基于分别与第一属性对应的第一原始人脸样本图像和第一风格人脸样本图像、以及分别与第二属性对应的第二原始人脸样本图像和第二风格人脸样本图像训练得到；第一风格人脸样本图像基于第一人脸图像生成模型的第一输出图像得到；第二风格人脸样本图像基于第二人脸图像生成模型的第二输出图像得到。本公开实施例优化了风格图像生成模型的训练效果，保证了该模型对不同属性的图像的通用性，确保了模型生成的风格图像的展示效果。

技术领域

本公开涉及图像处理技术领域，尤其涉及一种风格图像生成方法、模型训练方法、装置和设备。

背景技术

目前，随着视频交互应用的功能逐渐丰富化，图像风格转换成为了一种新的趣味性玩法。图像风格转换是指将一幅或者多幅图像进行风格转换，生成符合用户需求的风格图像。

以漫画风格转换为例，即将用户图像转换为具有漫画风格的图像，现有的漫画风格图像生成方案中，通常由于模型训练效果的限制，导致模型输出的漫画风格图像展示效果欠佳。

发明内容

为了解决上述技术问题或者至少部分地解决上述技术问题，本公开实施例提供了一种风格图像生成方法、模型训练方法、装置和设备。

第一方面，本公开实施例提供了一种风格图像生成方法，包括：

获取原始人脸图像；

利用预先训练的风格图像生成模型，得到与所述原始人脸图像对应的目标风格图像；

其中，所述风格图像生成模型基于分别与第一属性对应的第一原始人脸样本图像和第一风格人脸样本图像、以及分别与第二属性对应的第二原始人脸样本图像和第二风格人脸样本图像训练得到；

所述第一风格人脸样本图像基于第一人脸图像生成模型针对所述第一原始人脸样本图像的第一输出图像得到，所述第一人脸图像生成模型基于分别与所述第一属性对应的第三原始人脸样本图像和标准风格人脸样本图像训练得到；

所述第二风格人脸样本图像基于第二人脸图像生成模型针对所述第二原始人脸样本图像的第二输出图像得到，所述第二人脸图像生成模型基于分别与所述第二属性对应的第四原始人脸样本图像和标准风格人脸样本图像训练得到；

所述第一属性或者所述第二属性用于区分样本图像上人脸的特征。

第二方面，本公开实施例还提供了一种风格图像生成模型训练方法，包括：

基于分别与第一属性对应的第三原始人脸样本图像和标准风格人脸样本图像，训练得到第一人脸图像生成模型；

将与所述第一属性对应的第一原始人脸样本图像作为所述第一人脸图像生成模型的输入，得到第一输出图像，并基于所述第一输出图像得到与所述第一属性对应的第一风格人脸样本图像；

基于分别与第二属性对应的第四原始人脸样本图像和标准风格人脸样本图像，训练得到第二人脸图像生成模型；

将与所述第二属性对应的第二原始人脸样本图像作为所述第二人脸图像生成模型的输入，得到第二输出图像，并基于所述第二输出图像得到与所述第二属性对应的第二风格人脸样本图像；

基于所述第一原始人脸样本图像和所述第一风格人脸样本图像、以及所述第二原始人脸样本图像和所述第二风格人脸样本图像，训练得到风格图像生成模型；所述风格图像生成模型用于得到与原始人脸图像对应的目标风格图像。

第三方面，本公开实施例还提供了一种风格图像生成装置，包括：

原始人脸图像获取模块，用于获取原始人脸图像；