Skip to content

clementabary/travelgan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image-to-image domain translation using TraVeLGAN

PyTorch implementation of TraVeLGAN: Image-to-image Translation by Transformation Vector Learning.

Add configuration file and run with :

python train.py --hparams=config_file --log=/runs/exp --device={0,1,..}

You can run the training of TraVeLGAN between two classes of CIFAR10 dataset using the cifar.json file. Download the dataset into the data folder and run the command above.

Here is an example with bird-ship (32x32) translation after 500 epochs.

bird bird_ship

ship ship_bird

Here is an example with rock beauty-toucan (128x128) translation after 10k iterations (batch size 16).

bird bird_ship

ship ship_bird

As a side note, I could not reproduce results (persistent mode collapse) without spectral normalization.

About

PyTorch implementation of TraVeLGAN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages