OpenSTL Release V0.1.0 (SimVPv2)
Release version to SimVPv2 V0.1.0. This version is also known as OpenSTL V0.1.0. We have separated a branch for SimVPv2
since we plan to refactor this project with advanced features.
[2023-05-06] We add some meta files in this release to facilitate downloading.
[2023-06-04] We release the test sets of MMNIST datasets. Please refer to install.md for dataset preparation
in OpenSTL.
[2023-06-26] We add kth_action.zip
in the image format (.jpg
) for the kth dataloader.
Features
- Refactor code structures as
simvp/api
,simvp/core
,simvp/datasets
,simvp/methods
,simvp/models
,simvp/modules
. We support non-distributed training and evaluation by the executable python filetools/non_dist_train.py
. Refactor config files for SimVP models. - Fix bugs in tools/nondist_train.py, simvp/utils, environment.yml, and .gitignore, etc.
- Support Timm optimizers and schedulers.
- Update popular Metaformer models as the hidden Translator
$h$ in SimVP, supporting ViT, Swin-Transformer, MLP-Mixer, ConvMixer, UniFormer, PoolFormer, ConvNeXt, VAN, HorNet, and MogaNet. - Update implementations of dataset and dataloader, supporting KTH Action, KittiCaltech Pedestrian, Moving MNIST, TaxiBJ, and WeatherBench.
Documents
- Upload
readthedocs
documents. Summarize video prediction benchmark results on MMNIST in video_benchmarks.md. - Update benchmark results of video prediction baselines and MetaFormer architectures based on SimVP on MMNIST, TaxiBJ, and WeatherBench datasets.
- Update README and add a license.