写在前面的话
前几天看到了这篇博客,感觉很有意思就复现了一下,中途碰到一些问题跟大家分享一下。
项目获取
git clone https://github.com/bryandlee/animegan2-pytorch
或者直接在github上下载其压缩包保存下来
环境配置
我用的是pycharm社区版+pytorch+cuda+cudnn+anaconda。
具体流程:
1、下载并且安装anaconda,网上有很多教程
2、创建新的虚拟环境
3、下载自己电脑对应版本的cuda和cudnn
4、去pytorch官网获取对应版本的pytorch安装命令
5、打开anaconda promt输入activate [你的环境名称],进入后输入安装pytorch的命令。
6、然后进入anaconda navigator找到新环境,安装opencv
7、安装pycharm社区版,在里面打开文件
最终配置情况:
运行
原博客里面说的是直接输入
python test.py --checkpoint [模型文件路径] --input_dir [输入图像所在目录] --output_dir [输出目录] --device [设备选择,cpu或者cuda]
这里有几个问题
1、图片大小问题,原博客没有提及。代码没有对图像进行处理的过程,我一开始以为有,结果直接爆显存。图片越大,需要的现存越多。所以在输入图片的时候需要裁剪(如果你有钱就不用)。如果还是爆显存就用cpu吧。
2、再者是输入命令的目录问题,这里的输入图像目录是一个文件夹不是一个图像。
3、模型文件路径源码中的默认路径是不存在的,可以改一下所有的默认路径。权重文件有四个,大家看喜欢哪个用哪个。
推荐大家先改一下默认跑一下,每次输入命令报错挺麻烦的。
结果
原图片:
处理后的图片:
总结
这个还挺好玩的,后续会对视频做一个处理,对代码进行一个优化。大家有什么问题可以在评论区留言。