tmits37 / SimVPv2

The official implementation of the paper SimVP: Towards Simple yet Powerful Spatiotemporal Predictive learning.

SimVP: Towards Simple yet Powerful Spatiotemporal Predictive learning

This repository contains the implementation code for paper:

SimVP: Towards Simple yet Powerful Spatiotemporal Predictive learning
Cheng Tan, Zhangyang Gao, Stan Z. Li.

Introduction

This is the journal version of our previous conference work (SimVP: Simpler yet Better Video Prediction, In CVPR 2022).

The overall framework of SimVP.

The performance of SimVPs on the Moving MNIST dataset. For the training time, the less the better. For the inference efficiency (frames per second), the more the better.

Quantitative results of different methods on the Moving MNIST dataset ($10 \rightarrow 10$ frames).

Dependencies

• torch
• scikit-image=0.16.2
• numpy
• argparse
• tqdm

Overview

• API/ contains dataloaders and metrics.
• main.py is the executable python file with possible arguments.
• model.py contains the SimVP model.
• exp.py is the core file for training, validating, and testing pipelines.

Install

This project has provided an environment setting file of conda, users can easily reproduce the environment by the following commands:

  conda env create -f environment.yml
  conda activate SimVP

Moving MNIST dataset

  cd ./data/moving_mnist
  bash download_mmnist.sh

Citation

If you are interested in our repository and our paper, please cite the following paper:

@article{tan2022simvp,
  title={SimVP: Towards Simple yet Powerful Spatiotemporal Predictive Learning},
  author={Tan, Cheng and Gao, Zhangyang and Li, Stan Z},
  journal={arXiv preprint arXiv:2211.12509},
  year={2022}
}

Contact

If you have any questions, feel free to contact us through email ([e-mail]). Enjoy!