[发明专利]聚焦主体的双阶段彩墨风格图像生成方法及可读存储介质

说明书

本申请公开一种聚焦主体的双阶段彩墨风格图像生成方法，其包括：获取待生成彩墨风格图像的原始图像；对原始图像进行前背景分离，获取原始图像的前景图像B1和背景图像F1；将背景图像B1的风格迁移为指定的彩墨风格，生成彩墨画风格的背景图像B2；对前景图像进行细节增加处理，将细节增强后的前景图像的风格转移为指定的彩墨风格，生成彩墨风格的前景图像F2；以及融合背景图像B2和前景图像F2，获得最终的彩墨风格图像C。本申请还提供一种计算机可读存储介质。

技术领域

本申请涉及图像处理技术，更具体地说，涉及一种聚焦主体的双阶段彩墨风格图像生成方法及可以存储介质。

背景技术

风格迁移是指将一幅图像的风格转移到另一幅图像上以生成该风格图像的方法。传统上，这个问题主要通过纹理合成和纹理迁移来完成，虽然有一定的效果，但目标图像的质量仍不能满意。彩墨画是中国传统文化的传承，比起水墨画，它更具艺术性，同时也是一种文化自信的体现。近年来，通过风格转移将图像转移成彩墨风格图像的研究受到研究者的广泛关注，现有的一些风格迁移方法通过对全局图像进行风格化，生成效果不自然，前景目标主体不明确。

发明内容

针对现有技术，本申请解决的技术问题是提供一种实现彩墨风格转移的凸显前景目标主体的聚焦主体的双阶段彩墨风格图像生成方法。

为解决上述技术问题，本申请提供一种聚焦主体的双阶段彩墨风格图像生成方法，其包括：

获取待生成彩墨风格图像的原始图像；

对原始图像进行前背景分离，获取原始图像的前景图像B1和背景图像F1；

将背景图像B1的风格迁移为指定的彩墨风格，生成彩墨画风格的背景图像B2；

对前景图像进行细节增加处理，将细节增强后的前景图像的风格转移为指定的彩墨风格，生成彩墨风格的前景图像F2；以及

融合背景图像B2和前景图像F2，获得最终的彩墨风格图像C。

在一申请实施例中，将细节增强后的前景图像的风格转移为指定的彩墨风格，生成彩墨风格的前景图像F2的步骤包括：

分离细节增强后的前景图像的轮廓和纹理，获得轮廓图像L1和纹理图像W1；

将轮廓图像L1的风格迁移为指定的彩墨风格，获得轮廓彩墨风格图像L2；

将纹理图像W1的风格迁移为指定的彩墨风格，获得纹理彩墨风格图像W2；

融合轮廓彩墨风格图像L2和纹理彩墨风格图像W2，获得彩墨风格的前景图像F2。

在一申请实施例中，将轮廓图像L1的风格迁移为指定的彩墨风格或将纹理图像W1的风格迁移为指定的彩墨风格的步骤包括：

训练集准备：收集普通非彩墨风格的图像样本组成X域图像，收集指定的彩墨风格的图像样本组成Y域图像；

构建第一风格迁移网络：以卷积神经网络模型作为第一生成对抗网络的第一生成器G_X2Y和第一判别器D_Y，以X域图像中的每一图像x作为第一生成器G_X2Y的输入，以第一生成器G_X2Y的输出和Y域图像中的每一图像y作为第一判别器D_Y的输入；以卷积神经网络模型作为第二生成对抗网络的第二生成器G_Y2X和第二判别器D_X，以Y域图像作为第二生成器G_Y2X的输入，以第二生成器G_Y2X的输出和X域图像作为第二判别器D_X的输入；

构建第一风格迁移网络的损失函数及训练第一风格迁移网络：