在使用cyclegan网络运行自己数据集时遇到的一些问题及解决方法汇总

 cyclegan作为GANs的经典网络之一,其实用性和在图像风格转化上的效果是毋庸置疑的。但是该网络在使用时由于其庞大的网络架构,一般的电脑性能是运行不了,故不少人在使用该网络时会选择使用云服务器去跑代码,故本文章会将本人在运行此代码时的问题以及解决方法进行总结。

1.关于路径问题和参数设置问题

1.1由于不少人会使用自己的数据集进行测试,但是苦于找不到数据集修改的路径(虽然这个路径就知道了也不好修改),其实路径就是pytorch-CycleGAN-and-pix2pix\data\unaligned_dataset.,具体的修改的地方就是右手箭头那里。

1.2设置参数

设置参数主要在options内进行具体如下:

 其中修改计算机运行核心数是在--num_threads上进行修改,一般默认为4.。在训练前的准备工作一般就只有这些需要注意。

 2.服务器上训练的问题及解决方法

在服务器上训练的布置方法各路大神已经发了很多优秀的帖子了,大家自行前往csdn上寻找。这里我只讲自己遇到的问题。

2.1我是在autodl上租的服务器,在镜像环境那一块很难选到和自己一致的环境,我的经验是在pytorch和cuda一样的情况下,只要python大于等于自己电脑上安装的python环境就行。

2.2当大家的大环境搭建好后就需要进入自己搭建的环境,我的进入环境主要步骤如下(在终端输入以下代码):

1.进入自己的环境,这里x9py38是有个大佬建的环境的名字,我这里直接照搬了,输入后直接按回车。

conda activate x9py38

2. 输入ls查看该目录下存在有自己所需要文件夹不

 3.打开自己所需文件夹

cd autodl-tmp/pytorch-CycleGAN-and-pix2pix

4.这里所遇到的问题令我困扰了许久,到这一步一般需要输入python train.py 然后运行,但是我遇到问题, 具体如下:

环境会告诉我们dataroot出现了问题,当遇到这个问题的时候我寻求去代码本身去修改路径,但想遍办法也这么都不对,突然想到是否是需要在环境中指定路径呢?说干就干!

 python train.py --dataroot ./datasets/maps

没想到使用这个代码真的解决了这个问题,后面maps是你数据集的名字,大家在使用时候修改成自己的数据集名字。

有时候还会提醒你有numpy的版本问题这里我的经验是重新安装,这里安装时候如果pip install Numpy不行那就切换conda install Numpy指令。 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值