0
点赞
收藏
分享

微信扫一扫

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


先看下效果图:

左边是原图,右边是处理后的图片,使用的 face_paint_512_v2 模型。

Python图像卡通化animegan2-pytorch实例演示_深度学习


项目获取:

animegan2-pytorch 下载解压后 cmd 可进入项目地址的命令界面。

其中 img 是我自己建的,用于存放图片。

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


需要 torch 版本 >= 1.7.1

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


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

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


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

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


--input_dir 参数指向为要处理的图片所在目录。

--output_dir 参数指向为处理后的图片输出目录。

如果不指定输出目录的话,默认会存放在 samples\results目录下。

Python图像卡通化animegan2-pytorch实例演示_深度学习_06


--device 参数是设备选择,可以选择 cpucuda

对另一个图片也进行处理

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


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

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


喜欢的点个赞❤吧!


举报

相关推荐

0 条评论