README.md 816 字节
Newer Older
1 2
English | [简体中文](./README.md)

L
LielinJiang 已提交
3 4 5 6
# PaddleGAN

still under development!!

7 8 9 10 11 12 13
## 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]
```
L
LielinJiang 已提交
14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
## 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
```