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

权利要求书

1.一种图像风格迁移方法，包括：

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

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

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

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

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

2.根据权利要求1所述的方法，其中，所述利用所述第一下采样图像和所述第二下采样图像，生成初步风格迁移图像，包括：

对所述第一下采样图像进行编码，得到第一图像特征；并对所述第二下采样图像进行编码，得到第二图像特征；

将所述第一图像特征和所述第二图像特征进行组合，得到组合特征；

对所述组合特征进行解码，得到所述初步风格迁移图像。

3.根据权利要求1或2所述的方法，其中，所述第一上采样图像中的数据采用大小为3*W1*H1的第一矩阵表示，所述第一矩阵中的各个元素分别表示所述第一上采样图像中各个像素的三原色；

所述拉普拉斯金字塔图像中的数据采用大小为3*W1*H1的第二矩阵表示，所述第二矩阵中的各个元素分别表示所述拉普拉斯金字塔图像中各个像素的三原色；

其中，所述W1表示所述第一上采样图像的宽度，或者所述拉普拉斯金字塔图像的宽度；所述H1表示所述第一上采样图像的高度，或者所述拉普拉斯金字塔图像的高度；

所述将所述第一上采样图像中的数据与所述内容图像的拉普拉斯金字塔图像中的数据进行组合，得到组合后的数据，包括：将所述第一矩阵中的数据和所述第二矩阵中的数据组合，得到组合后的数据；所述组合后的数据采用大小为6*W1*H1的第三矩阵表示。

4.根据权利要求1或2所述的方法，其中，生成所述内容图像的拉普拉斯金字塔图像，包括：

将所述内容图像进行下采样，得到所述第一下采样图像；

将所述第一下采用图像进行上采样，得到第二上采样图像；

计算所述第二上采样图像与所述内容图像的差异；

利用所述差异生成所述内容图像的拉普拉斯金字塔图像。

5.根据权利要求4所述的方法，其中，所述第二上采样图像中的数据采用大小为3*W2*H2的第四矩阵表示，所述第四矩阵中的各个元素分别表示所述第二上采样图像中各个像素的三原色；

所述内容图像中的数据采用大小为3*W2*H2的第五矩阵表示，所述第五矩阵中的各个元素分别表示所述内容图像中各个像素的三原色；

其中，所述W2表示所述第二上采样图像的宽度，或者所述内容图像的宽度；所述H2表示所述第二上采样图像的高度，或者所述内容图像的高度；

所述计算所述第二上采样图像与所述内容图像的差异，包括：计算所述第四矩阵与所述第五矩阵的差，得到大小为3*W2*H2的第六矩阵，所述第六矩阵表示所述第二上采样图像与所述内容图像的差异。

6.一种图像风格迁移装置，包括：

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

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

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

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

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