[发明专利]风格迁移方法、装置、存储介质及电子设备

说明书

本申请实施例公开了一种风格迁移方法、装置、存储介质及电子设备，通过获取需要进行风格迁移的内容图像，以及获取用于风格迁移参考的写实图像，以及对内容图像进行语义分割，得到第一语义分割结果，以及对写实图像进行语义分割，得到第二语义分割结果，最后基于第一语义分割结果和第二语义分割结果，调用预训练的风格迁移模型将写实图像的写实风格迁移到内容图像，得到写实内容图像。由此，本申请能够实现对图像写实风格的迁移。

技术领域

本申请涉及图像处理技术领域，具体涉及一种风格迁移方法、装置、存储介质及电子设备。

背景技术

区别于颜色转换，风格迁移是一种将一张图像中的色彩、形态、纹理等高层级的抽象特征迁移到另一张图像上，使得另一张图像的视觉风格与之相似的技术。相关技术中，往往普遍关注于艺术类图像的风格迁移，从结果上看，艺术类图像的风格迁移更集中在艺术效果的转换，对图像信息的保真较低，结果图像往往偏向艺术化、抽象化。而对于写实风格的迁移，则鲜有关注。

发明内容

本申请实施例提供了一种风格迁移方法、装置、存储介质及电子设备，能够实现对图像写实风格的迁移。

本申请实施例提供的风格迁移方法，包括：

获取需要进行风格迁移的内容图像；

获取用于风格迁移参考的写实图像；

对所述内容图像进行语义分割，得到第一语义分割结果，以及对所述写实图像进行语义分割，得到第二语义分割结果；

基于所述第一语义分割结果和所述第二语义分割结果，调用预训练的风格迁移模型将所述写实图像的写实风格迁移到所述内容图像，得到写实内容图像。

本申请实施例提供的风格迁移装置，包括：

第一获取模块，用于获取需要进行风格迁移的内容图像；

第二获取模块，用于获取用于风格迁移参考的写实图像；

语义分割模块，用于对所述内容图像进行语义分割，得到第一语义分割结果，以及对所述写实图像进行语义分割，得到第二语义分割结果；

风格转换模块，用于基于所述第一语义分割结果和所述第二语义分割结果，调用预训练的风格迁移模型将所述写实图像的写实风格迁移到所述内容图像，得到写实内容图像。

本申请实施例提供的存储介质，其上存储有计算机程序，当所述计算机程序被处理器加载时执行如本申请提供的风格迁移方法。

本申请实施例提供的电子设备，包括处理器和存储器，所述存储器存有计算机程序，所述处理器通过加载所述计算机程序，用于执行本申请提供的风格迁移方法。

本申请通过获取需要进行风格迁移的内容图像，以及获取用于风格迁移参考的写实图像，以及对内容图像进行语义分割，得到第一语义分割结果，以及对写实图像进行语义分割，得到第二语义分割结果，最后基于第一语义分割结果和第二语义分割结果，调用预训练的风格迁移模型将写实图像的写实风格迁移到内容图像，得到写实内容图像。由此，本申请能够实现对图像写实风格的迁移。

附图说明

为了更清楚地说明本申请实施例中的技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本申请的一些实施例，对于本领域技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。

图1为本申请实施例提供的风格迁移方法的流程示意图。

图2是本申请实施例提供的风格迁移界面的示例图。

图3是本申请实施例中进行风格迁移的示例图。

图4是本申请实施例提供的风格迁移模块的结构示意图。