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

Posted 2023-03-03 挣扎的蓝藻

先看下效果图：
左边是原图，右边是处理后的图片，使用的 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 模型处理后的好一点。

喜欢的点个赞❤吧！