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

说明书

本申请实施例提供了图像风格迁移及数据存储方法、装置和电子设备，应用于图像处理技术领域，该数据存储方法包括：将多种风格类型的图像输入到预设卷积神经网络模型中进行训练，得到中间特征图及各风格类型滤波器的组输出激活相对于中间特征图的权重值；存储各风格类型滤波器组的输出激活相对于中间特征图的权重值及中间特征图。在本申请实施例的数据存储方法中，针对多种风格类型，存储各风格类型的滤波器组的输出激活相对于中间特征图的权重值及中间特征图，而权重值的数据量远远小于特征图的数据量的，因此相比于针对每个风格类型的滤波器组单独存储一个特征图，可以降低存储空间的占用量，节约存储空间。

技术领域

本申请涉及图像处理技术领域，特别是涉及图像风格迁移及数据存储方法、装置和电子设备。

背景技术

Style Transfer(风格迁移)是近年来在图像处理及计算机视觉领域中备受关注，并且具有实际应用价值的技术。风格迁移可以将一张Original Image(原始图像)，转变成一张经过艺术加工的Stylized Image(风格化图像)。其中风格元素来自于另一张称为Style Image(风格图像)的图像。一个典型的例子是将任意一张图像变为拥有梵高著名油画《星空》风格的图像。

在基于Style Bank(风格分离)框架的风格迁移方法中，不同的风格是通过中间的128维的Filter Bank(滤波器组)来表示的，每一种风格都有自己的滤波器组，需要分别存储各风格类型的录波器组的输出激活的特征图，因此当风格越来越多，存储的数据将线性增长，占用大量存储空间。

发明内容

本申请实施例的目的在于提供一种图像风格迁移及数据存储方法、装置和电子设备，以实现降低存储空间的占用量。具体技术方案如下：

第一方面，本申请实施例提供了一种数据存储方法，所述方法包括：

将多种风格类型的图像输入到预设卷积神经网络模型中进行训练，得到中间特征图及各所述风格类型滤波器的组输出激活相对于所述中间特征图的权重值；

存储各所述风格类型滤波器组的输出激活相对于所述中间特征图的权重值及所述中间特征图。

可选的，本申请实施例的数据存储方法基于风格分离框架。

可选的，在所述存储各所述风格类型滤波器组的输出激活相对于所述中间特征图的权重值及所述中间特征图之后，所述方法还包括：

获取待风格迁移图像、待转换的目标风格类型及所述中间特征图；

按照所述目标风格类型，获取所述目标风格类型的滤波器组的输出激活相对于所述中间特征图的目标权重值；

按照所述目标权重值，调整所述中间特征图中各特征的权重，得到目标滤波器组输出激活；

通过所述目标滤波器组输出激活对所述待风格迁移图像进行处理，得到风格迁移后的图像。

第二方面，本申请实施例提供了一种图像风格迁移方法，所述方法包括：

获取待风格迁移图像、待转换的目标风格类型及预设中间特征图；

按照所述目标风格类型，获取所述目标风格类型的滤波器组的输出激活相对于所述中间特征图的目标权重值；

按照所述目标权重值，调整所述中间特征图中各特征的权重，得到目标滤波器组输出激活；

通过所述目标滤波器组输出激活对所述待风格迁移图像进行处理，得到风格迁移后的图像。

可选的，本申请实施例的图像风格迁移方法基于风格分离框架。

第三方面，本申请实施例提供了一种数据存储装置，所述装置包括：