Awesome

IntroVAE

A pytorch implementation of Paper "IntroVAE: Introspective Variational Autoencoders for Photographic Image Synthesis"

Prerequisites

• Python 2.7 or Python 3.6
• PyTorch

Run

Use the default parameters except changing the hyper-parameter, i.e., m_plus, weight_rec, weight_kl, and weight_neg, for different image resolution settings. Noted that setting num_vae nonzero means pretraining the model in the standard VAE manner, which may helps improve the training stablitity and convergency.

The default parameters for CelebA-HQ faces at 256x256 and 1024x1024 resolutions are provided in the file 'run_256.sh' and 'run_1024.sh', respectively. Other settings are allowed as discussed in the appendix of the published paper.

Results

Citation

If you use our codes, please cite the following paper:

@inproceedings{huang2018introvae,
  title={IntroVAE: Introspective Variational Autoencoders for Photographic Image Synthesis},
  author={Huang, Huaibo and Li, Zhihang and He, Ran and Sun, Zhenan and Tan, Tieniu},
  booktitle={Advances in Neural Information Processing Systems},
  pages={10236--10245},    
  year={2018}
}

The released codes are only allowed for non-commercial use.