开源人像分割

昨天逛gayhub，又看到一个有趣的开源项目，很有趣，python源码也值得学习，有依赖一些深度学习的库。

人像卡通风格渲染的目标是，在保持原图像ID信息和纹理细节的同时，将真实照片转换为卡通风格的非真实感图像。

人像卡通化 Photo to Cartoon

1.安装依赖库

项目所需的主要依赖库如下：

2.下载资源

资源地址：关注后私信“cartoon”获取(我只是为了涨粉啦)，或者查看评论置顶。

人像卡通化预训练模型：photo2cartoon_weights.pt(20200504更新)，存放在models路径下。

头像分割模型：seg_model_384.pb，存放在utils路径下。

人脸识别预训练模型：model_mobilefacenet.pth，存放在models路径下。(From: InsightFace_Pytorch)

卡通画开源数据：cartoon_data，包含trainB和testB。

3.测试

将一张测试照片(亚洲年轻女性)转换为卡通风格：

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

大致效果如下：

卡通化效果

1.数据准备

训练数据包括真实照片和卡通画像，为降低训练复杂度，我们对两类数据进行了如下预处理：

检测人脸及关键点。

根据关键点旋转校正人脸。

将关键点边界框按固定的比例扩张并裁剪出人脸区域。

使用人像分割模型将背景置白。

我们开源了204张处理后的卡通画数据，您还需准备约1000张人像照片(为匹配卡通数据，尽量使用亚洲年轻女性照片，人脸大小最好超过200x200像素)，使用以下命令进行预处理：

将处理后的数据按照以下层级存放，trainA、testA中存放照片头像数据，trainB、testB中存放卡通头像数据。

2.训练

重新训练:

加载预训练参数:

我认为这个开源项目对于想学习 Python 和学习 Python相关人工智能项目的人来讲是一个不错的学习和研究的资料，看看人家是怎么搞的，学习人家的思想和思路。

欢迎关注技术阁，专注于分享科技资讯和技术干货。