纪念第一跑完了junyanz/pytorch-CycleGAN-and-pix2pix。首先说,我是个纯小白,这也是个纯手工活的过程,纪录一下我踩过的坑。
我的电脑是:6GB的GPU,windows10系统,pytorch
一 git clone pytorch-CycleGAN-and-pix2pix
开始键+r打开运行,输入cmd打开
我将clone的文件放入了F盘,所以在cmd种输入f:,回车后输入git clone +https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix.git.这样就保存到了f盘。
二 按照readme文档进行
1 下载数据集
这个里面的数据集打不开,建议换一个,命名为trainA和trainB,保存到datasets下的maps文件夹里。
2 可视化
打开cmd,输入f:,回车后输入下载文件的路径,再回车,输入python -m visdom.server,用浏览器打开 http://localhost:8097.
3 开始训练
打开心得cmd,输入f:,回车后输入下载文件的路径,再回车,输入
python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan
训练完之后别关cmd
4 训练完之后,开始测试,将测试文件夹命名为testA和testB,保存在maps文件夹下。
在训练完的cmd种输入
python test.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan
在reslut文件夹里就能看到结果了。
三 遇到的问题
1 有些论文的数据集可以下载,是sh文件,下载一个git,就能将对应的sh文件打开并下载到数据集,数据集的压缩文件会下到浏览器里。
2 训练开始就显示 gpu内存不够,换成cpu就行,将程序里的0换为-1
如果遇到其他问题,欢迎留言,一起解决