Skip to content
/ tagan Public

An official PyTorch implementation of the paper "Text-Adaptive Generative Adversarial Networks: Manipulating Images with Natural Language", NeurIPS 2018

License

Notifications You must be signed in to change notification settings

woozzu/tagan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text-Adaptive Generative Adversarial Network (TAGAN)

A PyTorch implementation of the paper "Text-Adaptive Generative Adversarial Networks: Manipulating Images with Natural Language". This code implements a Text-Adaptive Generative Adversarial Network (TAGAN) for manipulating images with natural language.

Model architecture

Requirements

Pretrained word vectors for fastText

Download a pretrained English word vectors. You can see the list of pretrained vectors on this page.

Datasets

The caption data is from this repository. After downloading, modify CONFIG file so that all paths of the datasets point to the data you downloaded.

Pretrained models

Please put these files in ./models/ folder.

Run

  • scripts/preprocess_caption.sh
    Preprocess caption data using fastText embedding. You only need to run it once before training.
  • scripts/train_[flowers/birds].sh
    Train a network. If you want to change arguments, please refer to train.py.
  • scripts/test_[flowers/birds].sh
    Test a trained network. After running it, please see ./test/result_[flowers/birds]/index.html.

Results

Flowers

Birds

Citation

Please cite our paper when you use this code.

@inproceedings{nam2018tagan,
  title={Text-Adaptive Generative Adversarial Networks: Manipulating Images with Natural Language},
  author={Nam, Seonghyeon and Kim, Yunji and Kim, Seon Joo},
  booktitle={Advances in Neural Information Processing Systems (NeurIPS)},
  year={2018}
}

Contact

Please contact [email protected] if you have any question about this work.

About

An official PyTorch implementation of the paper "Text-Adaptive Generative Adversarial Networks: Manipulating Images with Natural Language", NeurIPS 2018

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published