Skip to content
master
Switch branches/tags
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

README.md

TuiGAN-PyTorch

Official PyTorch Implementation of "TuiGAN: Learning Versatile Image-to-Image Translation with Two Unpaired Images" (ECCV 2020 Spotlight)

TuiGAN's applications

TuiGAN can be use for various computer vision tasks ranging from image style transfer to object transformation and appearance transformation:

Usage

Install dependencies

python -m pip install -r requirements.txt

Our code was tested with python 3.6 and PyToch 1.0.0 or 1.2.0

Train

To train TuiGAN model on two unpaired images, put the first training image under datas/task_name/trainA and the second training image under datas/task_name/trainB, and run

python train.py --input_name <input_name> --root <datas/task_name>

For example,

python train.py --input_name apple --root datas/apple

Comparison Results

General Unsupervised Image-to-Image Translation

Image Style Transfer

Animal Face Translation

Painting-to-Image Translation

More Results

Art Style Transfer

Photorealistic Style Transfer

Animal Face Translation

About

PyTorch Implementation of ECCV 2020 Spotlight "TuiGAN: Learning Versatile Image-to-Image Translation with Two Unpaired Images"

Resources

Releases

No releases published

Packages

No packages published

Languages