先看下效果图:
左边是原图,右边是处理后的图片,使用的 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 模型处理后的好一点。
喜欢的点个赞❤吧!