不装了，我摊牌了！辰哥实现了人像卡通化的功能，玩了绝对上瘾！
今天咱们来搞点有趣的事情：人像卡通化。本文的人像卡通化功能没有使用第三方接口功能，辰哥知道百度有提供接口，并且十来行代码就可以搞定，但是有使用次数限制，所以辰哥就通过搭建神经网络模型，借助数据集进行训练，最后得到模型。
这样得到的人像卡通化的功能可以顺便使用，并且还可以通过提高数据集的质量或者调整参数，让生成的卡通人像更加逼真！
先来看一下效果：
看完效果之后，是不是觉得很神奇？辰哥告诉你，实际很简单，看完本文之后，你也可以自己动手直接生成你自己喜欢的动漫头像。
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这两个版本很很大区别）
从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