Awesome
RelGAN-PyTorch
A PyTorch implementation of RelGAN: Multi-Domain Image-to-Image Translation via Relative Attributes
The paper is accepted to ICCV 2019. We also have the Keras version here.
Get Started
Install
- Python 3.6 or higher
- PyTorch 0.4.0 or higher
- All the dependencies
pip3 install -r requirements.
Start TensorBoard server
tensorboard --logdir runs
Train your RelGAN!
CUDA_VISIBLE_DEVICES=0 python3 train.py --data <PATH_TO_CELEBA-HQ> --gpu [--image_size 256]
Use multiple GPUs
CUDA_VISIBLE_DEVICES=0,1,2,3 python3 train.py --data <PATH_TO_CELEBA-HQ> --multi_gpu [--image_size 256]
Specify your own training settings in config.yaml
CUDA_VISIBLE_DEVICES=0 python3 train.py --config config.yaml