github 人像卡通化探索项目

把项目下载到本地

• 下载地址

https://github.com/minivision-ai/photo2cartoon

安装依赖库

python 3.7 # 3.x版本都可
pytorch 1.4 
tensorflow-gpu 1.14 # tesorflow 得是1.0版本，2.0版本语法部分改变，不然项目运行会出错
face-alignment
dlib

• pytorch 安装

pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

更多请看

• 是否安装成功
  
• tensorflow-gpu 安装

有个dlib是特殊的，剩下的pip直接安装就行
里面有dlib安装

下载模型数据

https://pan.baidu.com/share/init?surl=MsT3-He3UGipKhUi4OcCJw
提取码:y2ch

• 解压里面有三个模型文件
  
• 把解压好的文件放到对应的路径下

1. 人像卡通化预训练模型：photo2cartoon_weights.pt，存放在models路径下。
2. 头像分割模型：seg_model_384.pb，存放在utils路径下。
3. 人脸识别预训练模型：model_mobilefacenet.pth，存放在models路径下。（From:
4. InsightFace_Pytorch） 卡通画开源数据：cartoon_data，包含trainB和testB。

• 如models路径下
  

启动项目测试一个小demo，图片动漫化

• 打卡cmd，切换到项目路径
• 运行下面的命令，我把photo_test.jsp换成jay了

python test.py --photo_path ./images/photo_test.jpg --save_path ./images/cartoon_result.png

• 开始转换成卡通图片

• 下面的是转换效果
  
  
  我去，hiahiahia这个有点像女生…因为这个项目的模型就是以亚洲女性来的。

• 最后，用图片训练就懒得搞了，没有图片…

• 再放一张女性的卡通图片