[发明专利]一种前后景分离的熊猫照片卡通化方法

说明书

本发明提供一种前后景分离的熊猫照片卡通化方法包括：获取熊猫照片数据集和卡通图像数据集并进行图像预处理：构建熊猫卡通图像生成模型；所述熊猫卡通图像生成模型包括生成器和判别器，所述生成器包括前景生成分支、后景生成分支和前后景合成分支；步骤4，定义所述生成器的损失函数；利用定义的损失函数训练所述熊猫卡通图像生成模型；将待卡通化的熊猫照片输入所述训练好的熊猫卡通图像生成模型，得到卡通化后的熊猫照片。相比于现有技术，本发明中(1)前后景分别处理的方法可以有效突出相对简洁的前景主体，(2)通过边缘增强和边缘模糊的处理能够得到边缘清晰的卡通化熊猫图像，(3)网络结构比较简单，在获得卡通效果的同时拥有较小的训练成本。

技术领域

本发明涉及图像处理技术领域，具体而言，涉及一种前后景分离的熊猫照片卡通化方法。

背景技术

近年来，人工智能算法逐渐被应用于图像生成领域，同时也为艺术作品的诠释带来了更大的可能性。对图像进行风格化时，图像在保留原图内容的同时，能够具备绘画作品的艺术风格，其创造性和艺术价值得到了人们的肯定。其中，基于生成对抗网络的方法(GAN)自2014年提出以来，越来越受到学术界和工业界的重视，不断在图像生成领域取得突破性进展，且仍然具有巨大潜力。它的对抗思想让生成器和判别器在博弈中互相进步，从而生成以假乱真的图片。卡通风格是动漫的一种，具有高度简化和抽象的特征，作为一种重要的艺术形式，无论是以漫画还是动画为载体，都具有独特的艺术效果，同时还体现角色本身的特点，深受人们的喜爱。若能结合生成对抗网络方法完成卡通图像的生成，则能大大降低人工成本。而卡通风格的图像生成效果一直不是很理想，这主要是因为现有方法大多没有针对卡通作品和其它艺术风格之间的显著区别进行探索：1、卡通风格的作品前景比背景更为简洁；2、卡通风格的作品具有平滑清晰且连贯的线条；3、卡通风格的作品色彩过渡平滑且纹理较为简单，而其它艺术风格大多拥有较为丰富的纹理和色彩变化。

现有方法虽然通过添加语义内容损失函数以及边缘损失函数实现了对纹理和色彩的部分简化，但均为全局的图像生成算法，并没有解决前景比背景更加简洁的问题。此外，现有算法虽然能够通驱使模型远离边缘模糊的负样本以生成比较清晰的边缘，但是对于熊猫等多毛发、边缘变化丰富的照片而言，生成的边缘线条不够平滑自然。

发明内容

本发明旨在提供一种前后景分离的熊猫照片卡通化方法，以解决前景不突出、边缘轮廓不清晰、色彩过渡不平滑以及纹理过多的问题。

本发明提供的一种前后景分离的熊猫照片卡通化方法，包括如下步骤：

步骤1，获取熊猫照片数据集和卡通图像数据集；

步骤2，对所述熊猫照片数据集和卡通图像数据集进行图像预处理：

步骤2.1，对所述熊猫照片数据集进行图像分割得到前景掩膜矩阵和后景掩膜矩阵；

步骤2.2，将所述熊猫照片数据集分别与前景掩膜矩阵和后景掩膜矩阵做与运算得到前景图像集和后景图像集；

步骤2.3，对所述前景图像集进行边缘增强得到边缘增强的前景图像集；

步骤2.4，对所述卡通图像数据集进行边缘模糊得到边缘模糊的卡通图像数据集；

步骤3，构建熊猫卡通图像生成模型；所述熊猫卡通图像生成模型包括生成器和判别器，所述生成器包括前景生成分支、后景生成分支和前后景合成分支；

步骤4，定义所述生成器的损失函数；

步骤5，利用定义的损失函数训练所述熊猫卡通图像生成模型：

步骤5.1，将所述前景图像集输入所述前景生成分支，输出前景图像特征；

步骤5.2，将所述后景图像集输入所述后景生成分支，输出后景图像特征；

步骤5.3，将前景图像特征和后景图像特征输入所述前后景合成分支，输出前后景聚合特征；