search

junyanz / pytorch-CycleGAN-and-pix2pix

Image-to-Image Translation in PyTorch
Other
23.09k stars 6.32k forks source link

How to train on the cityscapes dataset #1481

Open Lucky0775 opened 2 years ago

Lucky0775 commented 2 years ago

hi！Can you explain how to use the cycleGAN model to train the label (trainB) to image (trainA) task on the cityscape dataset and its training command?

junyanz commented 2 years ago

step 1: you need to download the datasets from the Cityscapes website: https://www.cityscapes-dataset.com/ step 2: In our experiments, we download the images to 128x128 or 256x256. You need to also put images and semantic label maps in different folders (trainA, trainB, testA, testB).
step 3: Follow the CycleGAN training and test README + change the --dataroot and --name accordingly.