[发明专利]图像生成方法、装置、设备及存储介质

说明书

本公开实施例公开了一种图像生成方法、装置、设备及存储介质。包括：获取包含目标人体的第一人体图像及包含目标衣物的第一衣物图像；对所述第一人体图像分别进行关键点提取、人像分割及人体部位分割，获得关键点特征图、人像分割图及人体部位分割图；将所述关键点特征图、所述人像分割图、所述人体部位分割图及所述第一衣物图像输入形变模型中，获得变形后的第二衣物图像；将所述第二衣物图像、第一人体图像、所述关键点特征图、所述人像分割图及所述人体部位分割图输入混合模型，获得第二人体图像；其中，所述第二人体图像中的所述目标人体穿戴所述目标衣物。本发明实施例提供的图像生成方法，可以提高生成图像的真实度。

技术领域

本公开实施例涉及图像处理技术领域，尤其涉及一种图像生成方法、装置、设备及存储介质。

背景技术

随着科技的发展，越来越多的应用软件走进了用户的生活，逐渐丰富了用户的业余生活，例如短视频APP等。用户可以采用视频、照片等方式记录生活，并上传到短视频APP上。

短视频APP上有许多基于图像算法与渲染技术的特效玩法。其中，虚拟换装是指应用图像融合技术，将用户的人体图像和包含目标衣物的衣物图像进行融合，得到用户穿戴该目标衣物后的图像，从而在用户无需真正试穿目标衣物的情况下，就能够了解到目标衣物的穿戴效果。

目前，在虚拟换装过程中，通常应用图像融合模型，分别对人体图像和衣物图像进行特征提取，基于提取到的两个图像特征生成新的图像，即用户穿戴目标衣物的图像。但是，在上述过程中，由于图像融合模型所提取的是粗略的图像特征，在生成图像时容易导致新生成的图像缺失细节信息，进而导致图像生成效果失真，虚拟换装的效果较差。

发明内容

本公开实施例提供一种图像生成方法、装置、设备及存储介质，可以提高生成图像的真实度。

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

获取包含目标人体的第一人体图像及包含目标衣物的第一衣物图像；

对所述第一人体图像分别进行关键点提取、人像分割及人体部位分割，获得关键点特征图、人像分割图及人体部位分割图；

将所述关键点特征图、所述人像分割图、所述人体部位分割图及所述第一衣物图像输入形变模型中，获得变形后的第二衣物图像；

将所述第二衣物图像、所述第一人体图像、所述关键点特征图、所述人像分割图及所述人体部位分割图输入混合模型，获得第二人体图像；其中，所述第二人体图像中的所述目标人体穿戴所述目标衣物。

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

人体图像获取模块，用于获取包含目标人体的第一人体图像及包含目标衣物的第一衣物图像；

分割图获取模块，用于对所述第一人体图像分别进行关键点提取、人像分割及人体部位分割，获得关键点特征图、人像分割图及人体部位分割图；

第二衣物图像获取模块，用于将所述关键点特征图、所述人像分割图、所述人体部位分割图及所述第一衣物图像输入形变模型中，获得变形后的第二衣物图像；

第二人体图像获取模块，用于将所述第二衣物图像、所述第一人体图像、所述关键点特征图、所述人像分割图及所述人体部位分割图输入混合模型，获得第二人体图像；其中，所述第二人体图像中的所述目标人体穿戴所述目标衣物。

第三方面，本公开实施例还提供了一种电子设备，所述电子设备包括：

一个或多个处理装置；

存储装置，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理装置执行，使得所述一个或多个处理装置实现如本公开实施例所述的图像生成方法。