search

liyunsheng13 / BDL

MIT License
222 stars 30 forks source link

Training CycleGAN #28

Open lrobberl opened 4 years ago

lrobberl commented 4 years ago

Hello, I'm trying to train the CycleGAN in order to use the same configuration you had while training the network, but using a different dataset. After following the readme.md file, I ran the code but there are several opt parameters inside the cycle_gan_model.py file of which I do not know the default parameters. Just to name few, there are opt.no_lsgan, opt.fineSize and so on... Could you please tell me what the default values for such parameters were, in order to retrain the CycleGAN for my dataset? My dataset has even a different amount of labels, so I cannot use the weights you provided for the segmentation model...

liyunsheng13 commented 4 years ago

opt.fineSize=1024 and opt.no_lsgan=False. I think when you said you have different amount of labels, it means the number of classes is different. In this case you should learn another segmentation model instead of using the one I upload.