Skip to content

Latest commit

 

History

History
41 lines (27 loc) · 902 Bytes

README.md

File metadata and controls

41 lines (27 loc) · 902 Bytes

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

  1. Python 3.6 or higher
  2. PyTorch 0.4.0 or higher
  3. 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