Python图像卡通化animegan2-pytorch实例演示

先看下效果图： 左边是原图，右边是处理后的图片，使用的 face_paint_512_v2 模型。

请添加图片描述

项目获取： animegan2-pytorch 下载解压后 cmd 可进入项目地址的命令界面。 其中 img 是我自己建的，用于存放图片。

在这里插入图片描述

需要 torch 版本 >= 1.7.1

在这里插入图片描述

然后目录下执行 python test.py --checkpoint .\weights\face_paint_512_v2.pt --input_dir .\img --device cpu 即可。

在这里插入图片描述

其中 --checkpoint 参数后面指向的为模型文件，一共有下面 4 种，每个模型处理图片后对应的样式也不一样。

在这里插入图片描述

--input_dir 参数指向为要处理的图片所在目录。 --output_dir 参数指向为处理后的图片输出目录。 如果不指定输出目录的话，默认会存放在 samples\results目录下。

在这里插入图片描述

--device 参数是设备选择，可以选择 cpu 或 cuda 对另一个图片也进行处理

请添加图片描述

处理后效果如下，同样感觉 face_paint_512_v2 模型处理后的好一点。

请添加图片描述