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

权利要求书

1.一种图像的风格迁移方法，其特征在于，包括步骤：

获取目标图像，接收风格选择指令，其中，所述风格选择指令包含：指定风格；

根据预置图像数据库，提取所述目标图像的特征，得到图像数组；

根据预置变换归一算法，对所述图像数组进行维度变换归一处理，得到变换数组；

根据预置张量转换算法，对所述变换数组进行张量转换处理，得到张量数组；

根据预置Pix2Pix神经网络，基于所述指定风格对所述张量数组进行特征迁移处理，得到风格化数组；

对所述风格化数组进行图像转换处理，生成风格迁移图像。

2.根据权利要求1所述的图像的风格迁移方法，其特征在于，在所述获取目标图像，接收风格选择指令之前，还包括：

获取N个训练样本图像，其中，N为正整数；

根据预置最近邻匹配算法，对所述N个训练样本图像进行风格分类处理，得到M个风格图像集，其中，M为小于N的正整数；

根据预置特征参数提取算法，对所述M个风格图像集进行风格特征提取处理，得到M个风格特征类别。

3.根据权利要求2所述的图像的风格迁移方法，其特征在于，所述根据预置特征参数提取算法，对所述M个风格图像集进行风格特征提取处理，得到M个风格特征类别包括：

将所述M个风格图像集输入预置pix2pix网络框架中，得到M个风格特征类别的Pix2Pix神经网络。

4.根据权利要求1所述的图像的风格迁移方法，其特征在于，所述根据预置图像数据库，提取所述目标图像的特征，得到图像数组包括：

根据预置图像数据库，对所述目标图像进行参数提取处理，得到BGR数组；

对所述BGR数组进行转换处理，得到RGB数组，以及将所述RGB数组确定为图像数组。

5.根据权利要求1所述的图像的风格迁移方法，其特征在于，所述根据预置变换归一算法，对所述图像数组进行维度变换归一处理，得到变换数组包括：

基于预置维度变换规则，对所述图像数组进行空间重排处理，得到维度重构数组；

将所述维度重构数组的元素均除以255，得到变换数组。

6.根据权利要求1所述的图像的风格迁移方法，其特征在于，所述根据预置张量转换算法，对所述变换数组进行张量转换处理，得到张量数组包括：

基于预置Pix2Pix神经网络的训练框架，对所述变换数组按元素排序拆解处理，得到张量数组。

7.根据权利要求1所述的图像的风格迁移方法，其特征在于，所述对所述风格化数组进行图像转换处理，生成风格迁移图像包括：

对所述风格化数组的所有元素均乘以255，得到像素数组；

基于所述像素数组，生成风格迁移图像。

8.一种图像的风格迁移装置，其特征在于，所述图像的风格迁移装置包括：

获取模块，用于获取目标图像，接收风格选择指令，其中，所述风格选择指令包含：指定风格；

特征提取模块，用于根据预置图像数据库，提取所述目标图像的特征，得到图像数组；

变换归一模块，用于根据预置变换归一算法，对所述图像数组进行维度变换归一处理，得到变换数组；

张量转换模块，用于根据预置张量转换算法，对所述变换数组进行张量转换处理，得到张量数组；

风格迁移模块，用于根据预置Pix2Pix神经网络，基于所述指定风格对所述张量数组进行特征迁移处理，得到风格化数组；

图像转换模块，用于对所述风格化数组进行图像转换处理，生成风格迁移图像。

9.一种图像的风格迁移设备，其特征在于，所述图像的风格迁移设备包括：存储器和至少一个处理器，所述存储器中存储有指令，所述存储器和所述至少一个处理器通过线路互连；

所述至少一个处理器调用所述存储器中的所述指令，以使得所述图像的风格迁移设备执行如权利要求1-7中任一项所述的图像的风格迁移方法。

10.一种计算机可读存储介质，所述计算机可读存储介质上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1-7中任一项所述的图像的风格迁移方法。