写在前面的话

前几天看到了这篇博客，感觉很有意思就复现了一下，中途碰到一些问题跟大家分享一下。

项目获取

代码源地址
可以下一个git bash把它克隆下来

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、模型文件路径源码中的默认路径是不存在的，可以改一下所有的默认路径。权重文件有四个，大家看喜欢哪个用哪个。

推荐大家先改一下默认跑一下，每次输入命令报错挺麻烦的。

结果

原图片：

处理后的图片：

总结

这个还挺好玩的，后续会对视频做一个处理，对代码进行一个优化。大家有什么问题可以在评论区留言。