2020-08-08

纪念第一跑完了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

如果遇到其他问题,欢迎留言,一起解决

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值