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

说明书

本公开提出一种图像风格迁移方法、装置、电子设备及存储介质，涉及计算机视觉和深度学习技术领域。具体实现方案为：生成内容图像的第一下采样图像、内容图像的拉普拉斯金字塔图像以及风格图像的第二下采样图像；利用第一下采样图像和第二下采样图像，生成初步风格迁移图像；对初步风格迁移图像进行上采样，得到第一上采样图像；将第一上采样图像中的数据与拉普拉斯金字塔图像中的数据进行组合，得到组合后的数据；对组合后的数据进行修正，得到修正后的图像；利用修正后的图像与初步风格迁移图像生成最终风格迁移图像。本公开实现的图像风格迁移能够达到较好的效果，且处理速度较快。

技术领域

本公开涉及人工智能技术领域，尤其涉及计算机视觉和深度学习技术领域。

背景技术

图像风格迁移又称图像风格转换、图像风格化处理等，是指将一张风格图像中的颜色和纹理风格迁移到一张内容图像上，同时保存内容图像的结构。

目前的图像风格化处理方法大体分为两种。第一种是优化方法，这类方法会采用固定网络的参数，通过优化图片本身来实现风格化；这种方法的缺点是处理速度慢，优化一张图片需要数分钟。第二种是前馈方法，这类方法在训练过程中训练风格化网络的参数，在测试时可以达到较快的速度；其缺点是纹理迁移不清晰，大尺度风格纹理无法迁移。

发明内容

本公开提供了一种图像风格迁移方法、装置、电子设备及存储介质。

根据本公开的一方面，提供了一种图像风格迁移方法，包括：

生成内容图像的第一下采样图像、内容图像的拉普拉斯金字塔图像以及风格图像的第二下采样图像；

利用第一下采样图像和第二下采样图像，生成初步风格迁移图像；

对初步风格迁移图像进行上采样，得到第一上采样图像；

将第一上采样图像中的数据与拉普拉斯金字塔图像中的数据进行组合，得到组合后的数据；

对组合后的数据进行修正，得到修正后的图像；利用修正后的图像与初步风格迁移图像生成最终风格迁移图像。

根据本公开的另一方面，提供了一种图像风格迁移装置，包括：

下采样模块，用于生成内容图像的第一下采样图像、内容图像的拉普拉斯金字塔图像以及风格图像的第二下采样图像；

初步生成模块，用于利用第一下采样图像和第二下采样图像，生成初步风格迁移图像；

处理模块，用于对初步风格迁移图像进行上采样，得到第一上采样图像；

组合模块，用于将第一上采样图像中的数据与拉普拉斯金字塔图像中的数据进行组合，得到组合后的数据；

修正模块，用于对组合后的数据进行修正，得到修正后的图像；利用修正后的图像与初步风格迁移图像生成最终风格迁移图像。

根据本公开的另一方面，提供了一种电子设备，包括：

至少一个处理器；以及

与该至少一个处理器通信连接的存储器；其中，

该存储器存储有可被该至少一个处理器执行的指令，该指令被该至少一个处理器执行，以使该至少一个处理器能够执行本公开任一实施例中的方法。

根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，该计算机指令用于使计算机执行本公开任一实施例中的方法。

根据本公开的另一方面，提供了一种计算机程序产品，包括计算机程序，该计算机程序被处理器执行时实现本公开任一实施例中的方法。

根据本公开的另一方面，提供了一种智能终端，包括如本公开的电子设备。