不装了，我摊牌了！人像卡通化的功能我实现了

不装了，我摊牌了！辰哥实现了人像卡通化的功能，玩了绝对上瘾！

今天咱们来搞点有趣的事情：人像卡通化。本文的人像卡通化功能没有使用第三方接口功能，辰哥知道百度有提供接口，并且十来行代码就可以搞定，但是有使用次数限制，所以辰哥就通过搭建神经网络模型，借助数据集进行训练，最后得到模型。

这样得到的人像卡通化的功能可以顺便使用，并且还可以通过提高数据集的质量或者调整参数，让生成的卡通人像更加逼真！

先来看一下效果：

看完效果之后，是不是觉得很神奇？辰哥告诉你，实际很简单，看完本文之后，你也可以自己动手直接生成你自己喜欢的动漫头像。

01、搭建环境

这里辰哥使用的是开源平台github的源码，里面有完整的模型结构、模型文件、数据集等。项目地址在下方

项目如下：
https://github.com/minivision-ai/photo2cartoon

可能看到这篇文章的读者没有深度学习的基础，但是没关系，辰哥一步一步教你如何将这个项目搭建起来，并且生成自己的卡通头像！

1.安装库

大家下载好源码之后，在运行之前，先把运行的环境搭建一下。

上面是项目中需要安装的库，实际上通过下面的四个命令安装即可（其中项目tensorflow-gpu说需要在gpu显卡下运行，但是实际上在咱们自己的笔记本或者电脑上运行也可以）

pip install onnxruntime
pip install face-alignment
pip install pytorch
pip install tensorflow==1.15

这里需要安装tensorflow版本是1，非2，不然就出现下面的报错（原因：1和2这两个版本很很大区别）

2.下载模型和数据集

从github上下载好代码之后，其目录结构如下：

这里给大家简单介绍一下相关文件夹和文件作用是什么？

文件夹：

dataset：存放训练数据集

images：存放测试数据集（预测模型效果所使用的图片存放文件夹）

models：训练好的模型存放在该目录下

utils：里面存放一些处理图片，模型结构等py文件

Py文件:

train.py：训练模型

test.py：测试模型（生成一张动漫人像图片）

主要就是这两个py文件需要去了解，即如何训练模型和使用模型（生成动漫人像图）

在项目中已提供了训练好的模型和数据集

在项目中还没有包含这些文件（已训练好模型、数据集等），需要我们再去下载，通过上面的下载地址，下载好之后再放入对应的文件夹即可。

这样我们就将这个项目的环境搭建完毕！

02、生成人像动漫图

1.训练模型

（已下载好了训练模型，想直接使用的可以跳过这一步，直接进行测试生成人像动漫图）

运行环境搭建完毕之后，下面可以开始进行训练模型。

先是对数据集进行预处理：

python data_process.py --data_path YourPhotoFolderPath --save_path YourSaveFolderPath

然后开始训练

python train.py --dataset photo2cartoon

如果是下载了预训练模型后，也可以在预训练模型的基础上进行训练

python train.py --dataset photo2cartoon --pretrained_weights models/photo2cartoon_weights.pt

训练结束之后的模型保存到models文件夹中。

2.测试生成人像动漫图

将需要生成动漫人像的原图放到images下：

执行生成命令：

# 使用默认模型
python test.py --photo_path ./images/lx.jpg --save_path ./images/cartoon_lx.png
# 使用onnx模型
python test_onnx.py --photo_path ./images/lx.jpg --save_path ./images/cartoon_lx.png

项目中提供了两种模型，因此在执行生成命令时也有两个（选择其中一条执行即可），其中原图片是：lx.jpg，生成的动漫人像是：cartoon_lx.png。

运行结果如下：

可以看到由一张真人人像图成功的生成了动漫人像图。

整个过程还是比较简单的，没有学过深度学习的小伙伴也可以将代码跑起来！！

03、小结

本文中教会了大家如何通过一张真人图去生成对应的动漫图，在文中也详细地讲解了如何去搭建环境，并代码跑起来。

鉴于可能很多读者不是学习深度学习方向的，所以文中就不太细于讲解代码，只是教会大家如何将代码跑起来，感兴趣的读者可以自己尝试！

一定要动手尝试 ！一定要动手尝试 ！一定要动手尝试！