Home

Awesome

Video Generation from Single Semantic Label Map

CVPR 2019 logoPaper accepted at 2019 IEEE Conference on Computer Vision and Pattern Recognition (CVPR)
Junting PanChengyu WangXu JiaJing ShaoLu ShengJunjie YanXiaogang Wang
Junting PanChengyu WangXu JiaJing ShaoLu ShengJunjie YanXiaogang Wang
<!--- A joint collaboration between: | ![logo-sensetime] | ![logo-huawei] | ![logo-cuhk] | |:-:|:-:|:-:| | [Sensetime Research][sensetime-web] | [Huawei][huawei-web] | [CUHK][cuhk-web] | [logo-sensetime]: https://github.com/junting/seg2vid/blob/junting/logos/sensetime.png "Sensetime Research" [logo-huawei]: https://github.com/junting/seg2vid/blob/junting/logos/huawei.png "Huawei" [logo-cuhk]: https://github.com/junting/seg2vid/blob/junting/logos/cuhk.png "cuhk" -->

Abstract

This paper proposes the novel task of video generation conditioned on a SINGLE semantic label map, which provides a good balance between flexibility and quality in the generation process. Different from typical end-to-end approaches, which model both scene content and dynamics in a single step, we propose to decompose this difficult task into two sub-problems. As current image generation methods do better than video generation in terms of detail, we synthesize high quality content by only generating the first frame. Then we animate the scene based on its semantic meaning to obtain the temporally coherent video, giving us excellent results overall. We employ a cVAE for predicting optical flow as a beneficial intermediate step to generate a video sequence conditioned on the initial single frame. A semantic label map is integrated into the flow prediction module to achieve major improvements in the image-to-video generation process. Extensive experiments on the Cityscapes dataset show that our method outperforms all competing methods.

Publication

Find our work on arXiv.

Image of the paper

Please cite with the following Bibtex code:

@article{pan2019video,
  title={Video Generation from Single Semantic Label Map},
  author={Pan, Junting and Wang, Chengyu and Jia, Xu and Shao, Jing and Sheng, Lu and Yan, Junjie and Wang, Xiaogang},
  journal={arXiv preprint arXiv:1903.04480},
  year={2019}
}

You may also want to refer to our publication with the more human-friendly Chicago style:

Junting Pan, Chengyu Wang, Xu Jia, Jing Shao, Lu Sheng, Junjie Yan and Xiaogang Wang. "Video Generation from Single Semantic Label Map." CVPR 2019.

Models

The Seg2Vid presented in our work can be downloaded from the links provided below the figure:

Seg2Vid Architecture architecture-fig

Img2Vid Architecture img2vid-fig

Visual Results

Cityscapes (Generation)

Generated Video 1Generated Video 2

Cityscapes (Prediction given the 1st frame and its segmetation mask)

Predicted Video 1Predicted Video 2
Predicted Flow 1Predicted Flow 2

Cityscapes 24 frames (Prediction given the 1st frame ans its segmetation mask)

Long Video 1Long Video 2Long Video 3

UCF-101 (Prediction given the 1st frame)

UCF Video 1UCF Video 2UCF Video 3UCF Video 4UCF Video 5UCF Video 6

KTH (Prediction given the 1st frame)

KTH Video 1KTH Video 2KTH Video 3KTH Video 4KTH Video 5

Getting Started

Please find our source code at this new repo here.

Dataset

Testing

  python -u test_refine_w_mask_two_path.py --suffix refine_w_mask_two_path --dataset cityscapes_two_path

Training

  python -u train_refine_multigpu_w_mask_two_path.py --batch_size 8 --dataset cityscapes_two_path

Seg2Vid on Pytorch

Seg2Vid is implemented in Pytorch.

Contact

If you have any general doubt about our work or code which may be of interest for other researchers, please use the public issues section on this github repo. Alternatively, drop us an e-mail at mailto:junting.pa@gmail.com.

<!--- Javascript code to enable Google Analytics -->