English | 简体中文
PaddleGAN
still under development!!
Download Dataset
This script could download several dataset for paired images for image2image translation task.
cd PaddleGAN/script/
bash pix2pix_download.sh [cityscapes|facades|edges2handbags|edges2shoes|maps]
Train
python -u tools/main.py --config-file configs/cyclegan-cityscapes.yaml
continue train from last checkpoint
python -u tools/main.py --config-file configs/cyclegan-cityscapes.yaml --resume your_checkpoint_path
multiple gpus train:
CUDA_VISIBLE_DEVICES=0,1 python -m paddle.distributed.launch tools/main.py --config-file configs/pix2pix-cityscapes.yaml
Evaluate
python tools/main.py --config-file configs/cyclegan-cityscapes.yaml --evaluate-only --load your_weight_path