Sliced Wasserstein Generative Models
-- and the application on unsupervised video generation
Papers
- Sliced Wasserstein Generative Models (CVPR2019)
- Towards High Resolution Video Generation with Progressive Growing of Sliced Wasserstein GANs (Arxiv)
Prerequisites
This repo has been successfully tested on tensorflow 1.10, cuda 9.0.
-
Please check the requirements.txt for more details.
-
For the training data such as Cifar10, CelebA, CelebA-HQ, LSUN etc download them on the official website accordingly.
-
TrailerFaces (Note that we tentatively release the tfrecord data to avoid the copyright issue.)
- The dataset contains approximately 200,000 individual clips of various facial expressions, where the faces are cropped with 256x256 resolution from about 6,000 high resolution movie trailers on YouTube. We convert them to tfrecord with resolutions range from 4x4 to 256x256. More about the data processing please see Towards high resolution video generation (Arxiv).
TrailerFaces sample:
Standardard Training
SWAE: it requires some custom ops, which are stored under the cuda folder.
- Following the instructions in install, you could compile them by yourself. If you install tensorflow by pip, one potential error can be some source files of tensorflow set the wrong relative path of cuda.h, you just need to manually change them according to your cuda path.
- Alternatively, you could also use the binary files directly, which is compiled with cuda8.0.
- Specify DATA_DIR, LOG_DIR and DIR in standard_training/swae_64x64.py, then run
- cd standard_training/
- python swae_64x64.py
SWGAN:
- Specify DATA_DIR, LOG_DIR and DIR in standard_training/swgan_64x64.py, then run
- cd standard_training/
- python swgan_64x64.py
Progressive Training
PG-SWGAN-3D:
-
Specify data_dir, result_dir in progressive_training/config.py, then run
- cd progressive_training/
- python train.py
-
We trained our model with 1 TitanXp GPU for roughly 7 days, since our code is built upon progressive growing GAN (https://arxiv.org/pdf/1710.10196.pdf), the code can be easily adapted to multigpu training on better GPU (See progressive_training/config.py). It is expected that the training speed can be significantly improved. Here are some sample frames generated by PG-SWGAN-3D:
-
More video comparison, see the following youtube links:
Citation
- If you use this code for your research, please cite our papers.
@inproceedings{jqwu&zwhuang2019swgm,
title={Sliced Wasserstein Generative Models},
author={Jiqing Wu, Zhiwu Huang, Dinesh Acharya, Wen Li, Janine Thoma, Danda Pani Paudel, Luc Van Gool},
booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
year={2019}
}
Acknowledgments
This code borrows from WGAN-GP (https://github.com/igul222/improved_wgan_training) and PGGAN (https://github.com/tkarras/progressive_growing_of_gans). We would like to thank them for the contribution.