Skip to content

RelGAN: Multi-Domain Image-to-Image Translation via Relative Attributes

Notifications You must be signed in to change notification settings

elvisyjlin/RelGAN-PyTorch

Repository files navigation

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

About

RelGAN: Multi-Domain Image-to-Image Translation via Relative Attributes

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages