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

权利要求书

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

获取实例样本集集合，所述实例样本集集合包括至少一个实例样本集，所述实例样本集包括至少一张样本图像，所述样本图像包括实例对象、实例属性，同一实例样本集中所述实例属性相同，所述实例样本集集合中的所述实例对象相同；

获取生成网络，所述生成网络包括判别器、生成器，所述判别器用于判别输入的图像是否为真实图像，所述生成器用于对图像进行风格迁移；

依据所述实例样本集集合对所述生成网络进行训练，获取训练完成后的所述生成器；

依据待迁移的目标实例获取所述生成器进行风格迁移后生成的结果图像。

2.根据权利要求1所述图像风格迁移方法，其特征在于，所述依据所述实例样本集集合对所述生成网络进行训练，获取训练完成后的所述生成器，包括：

依据所述实例样本集集合获取所述生成网络的第一损失函数值，所述第一损失函数为所述生成器的损失函数；

当所述第一损失函数值达到预设范围，获取训练完成后的所述生成器。

3.根据权利要求2所述图像风格迁移方法，其特征在于，所述依据所述实例样本集集合获取所述生成网络的第一损失函数值，包括：

获取所述判别器；

依据所述实例样本集集合、目标标签获取所述生成器生成的伪图像，所述目标标签为所述实例样本集集合中的所述实例对象迁移后具有的对象属性；

依据所述伪图像获取所述判别器的第一对抗损失；

依据所述伪图像、所述实例样本集集合中的所述实例对象的所述对象属性获取所述生成器的重建损失；

依据所述第一对抗损失、所述重建损失计算所述第一损失函数。

4.根据权利要求3所述图像风格迁移方法，其特征在于，获取所述判别器，包括：

依据所述实例样本集集合获取所述判别器的第二对抗损失及第一领域损失；

依据所述伪图像获取所述判别器的第三对抗损失；

在所述实例样本集集合及所述伪图像中采样获取采样样本；

依据所述采样样本获取所述判别器的梯度惩罚；

依据所述第二对抗损失、所述第三对抗损失、所述第一领域损失及所述梯度惩罚获取所述判别器的第二损失函数值；

依据所述第二损失函数值获取训练后的所述判别器。

5.根据权利要求1所述图像风格迁移方法，其特征在于，所述判别器、所述生成器为全卷积网络。

6.根据权利要求5所述图像风格迁移方法，其特征在于，所述判别器包括4个卷积层，前两个卷积层每层使用LeakyReLU激活函数，后两个卷积层无激活函数。

7.根据权利要求5所述图像风格迁移方法，其特征在于，所述生成器包括3个卷积层和1个转置卷积层，其中1个卷积层为残差模块。

8.根据权利要求1所述图像风格迁移方法，其特征在于，所述依据待迁移的目标实例获取所述生成器进行风格迁移后生成的结果图像，包括：

获取待迁移的目标图像中的所述目标实例及目标风格；

依据所述目标实例获取所述生成器按所述目标风格进行迁移的结果实例；

将所述结果实例替换所述目标图像中的所述目标实例，得到迁移后的所述结果图像。

9.一种图像风格迁移设备，其特征在于，所述图像风格迁移设备包括处理器、与所述处理器耦接的存储器，其中，

所述存储器存储有用于实现如权利要求1-8中任一项图像风格迁移方法的程序指令；

所述处理器用于执行所述存储器存储的所述程序指令以对图像风格进行迁移。

10.一种存储装置，其特征在于，存储有能够实现如权利要求1-8中任一项所述图像风格迁移方法的程序文件。