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

男娘i 2024-03-27 16:34 73阅读 0赞

**先看下效果图：**  
左边是原图，右边是处理后的图片，使用的 **face\_paint\_512\_v2** 模型。  
![请添加图片描述][e71ee26367034c5b8df2a0e648579837.png]  
**项目获取：**  
[**animegan2-pytorch**][animegan2-pytorch]  
下载解压后 **cmd** 可进入项目地址的命令界面。  
其中 **img** 是我自己建的，用于存放图片。  
![在这里插入图片描述][fb97ada250ef4a289709c25447c240ae.png]  
需要 **torch** 版本 `>= 1.7.1`  
![在这里插入图片描述][e77c7d86ccd64166b0bedb04e0e06747.png]  
然后目录下执行 `python test.py --checkpoint .\weights\face_paint_512_v2.pt --input_dir .\img --device cpu` 即可。  
![在这里插入图片描述][c539f66d676442c29e0de97c4f3d4798.png]  
其中 `--checkpoint` 参数后面指向的为模型文件，一共有下面 **4** 种，每个模型处理图片后对应的样式也不一样。  
![在这里插入图片描述][0b0c3a04688f421bb58f32d21b114f3f.png]  
`--input_dir` 参数指向为要处理的图片所在目录。  
`--output_dir` 参数指向为处理后的图片输出目录。  
如果不指定输出目录的话，默认会存放在 `samples\results`目录下。  
![在这里插入图片描述][23329ba6c2074fe4b21c5b537f640b4d.png]  
`--device` 参数是设备选择，可以选择 **cpu** 或 **cuda**  
对另一个图片也进行处理  
![请添加图片描述][cf96b7a7ae2447a38b9026b20b6e388a.png]  
处理后效果如下，同样感觉 **face\_paint\_512\_v2** 模型处理后的好一点。  
![请添加图片描述][74d2a79882174dea8b40691846caaad3.png]  
喜欢的点个赞❤吧！

[e71ee26367034c5b8df2a0e648579837.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/6def4fa00f6b4118878fe83912b8e85c.png
[animegan2-pytorch]: https://download.csdn.net/download/qq_38161040/87449155
[fb97ada250ef4a289709c25447c240ae.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/c10b5735dd024251b037fb9ecbabbd08.png
[e77c7d86ccd64166b0bedb04e0e06747.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/0fdf8d12972443f69add98438ff61876.png
[c539f66d676442c29e0de97c4f3d4798.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/2802a640aa12492188db8f74e8771f70.png
[0b0c3a04688f421bb58f32d21b114f3f.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/beeb5aa29b52485fad18a8519d088010.png
[23329ba6c2074fe4b21c5b537f640b4d.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/9724dd629a77404b860f5910b3520bb1.png
[cf96b7a7ae2447a38b9026b20b6e388a.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/c408048be9f54bc797a45dc36e13a2bf.png
[74d2a79882174dea8b40691846caaad3.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/27/29d93f98f19a44bca20668e9368e9d3e.png