sunpongber

【完结】看看开源项目

pytorch-CycleGAN-and-pix2pix

先看README

再看train
看一些参数,参数的设置是新手很困惑的地方,代码与README结合着看

pytorch-CycleGAN-and-pix2pix/options /train_options.py

README中有下面这段内容:
Train a model:

#!./scripts/train_cyclegan.sh
python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan

其实运行Python文件,我们在PyCharm中右键运行,其实除了这种方式外,我们可以通过在终端输入指令运行Python文件,比如python P30_test.py

那后面的参数是什么呢?--dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan
./datasets/maps等于给--dataroot赋值
maps_cyclegan等于给--name赋值
cycle_gan等于给--model赋值

那我们可以在pytorch-CycleGAN-and-pix2pix/options /train_options.py找一下--dataroot

没有找到?我们可以看到它是从class TrainOptions(BaseOptions):这个父类中继承

到这里pytorch-CycleGAN-and-pix2pix/options /base_options.py找一下
可以发现,这里有--dataroot

我们可以尝试把parser.add_argument('--dataroot', required=True, help='path to images (should have subfolders trainA, trainB, valA, valB, etc)')
修改为parser.add_argument('--dataroot', default=./datasets/maps, help='path to images (should have subfolders trainA, trainB, valA, valB, etc)')
然后在PyCharm中尝试运行

原始资料地址:
【完结】看看开源项目
如有侵权联系删除 仅供学习交流使用