【完结】看看开源项目
先看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中尝试运行
原始资料地址:
【完结】看看开源项目
如有侵权联系删除 仅供学习交流使用