Weather Prediction Visualization¶
We provide benchmark results of spatiotemporal prediction learning (STL) methods on famous weather prediction datasets, WeatherBench. More STL methods will be supported in the future. Issues and PRs are welcome! Visualization of GIF is released.
Table of Contents¶
Currently supported spatiotemporal prediction methods
[x] ConvLSTM (NeurIPS’2015)
[x] PredNet (ICLR’2017)
[x] PredRNN (NeurIPS’2017)
[x] PredRNN++ (ICML’2018)
[x] E3D-LSTM (ICLR’2018)
[x] MIM (CVPR’2019)
[x] CrevNet (ICLR’2020)
[x] PhyDNet (CVPR’2020)
[x] MAU (NeurIPS’2021)
[x] PredRNN.V2 (TPAMI’2022)
[x] SimVP (CVPR’2022)
[x] SimVP.V2 (ArXiv’2022)
[x] TAU (CVPR’2023)
[x] DMVFN (CVPR’2023)
Currently supported MetaFormer models for SimVP
[x] ViT (ICLR’2021)
[x] Swin-Transformer (ICCV’2021)
[x] MLP-Mixer (NeurIPS’2021)
[x] ConvMixer (Openreview’2021)
[x] UniFormer (ICLR’2022)
[x] PoolFormer (CVPR’2022)
[x] ConvNeXt (CVPR’2022)
[x] VAN (ArXiv’2022)
[x] IncepU (SimVP.V1) (CVPR’2022)
[x] gSTA (SimVP.V2) (ArXiv’2022)
[x] HorNet (NeurIPS’2022)
[x] MogaNet (ArXiv’2022)
We provide visualization figures of various weather prediction methods on Weather Bench (single variable). You can plot your own visualization with tested results (e.g., work_dirs/exp_name/saved
) by vis_video.py. Note that --vis_dirs
denotes visualize all experimental folders under the path, and --vis_channel
can select the channel for visualization. For example, run plotting with the script:
python tools/visualizations/vis_video.py -d weather_t2m_5_625 -w work_dirs/exp_name --index 0 --save_dirs fig_w_t2m_5_625_vis
WeatherBench Benchmarks Visualization¶
We provide temperature prediction benchmark results on the popular WeatherBench dataset (temperature prediction t2m
) using \(12\rightarrow 12\) frames prediction setting.
Visualization of STL Benchmarks on Temperature (t2m)¶
Similar to Moving MNIST Benchmarks, we benchmark popular STL methods training 50 epochs with single GPU on t2m
(K) in configs/weather/t2m_5_625 for 5.625
settings (\(32\times 64\) resolutions).
ConvLSTM |
---|
E3D-LSTM |
MAU |
---|---|
MIM |
PhyDNet |
---|---|
PredRNN |
PredRNN++ |
---|---|
PredRNN-V2 |
SimVP-V1 |
---|---|
SimVP-V2 |
TAU |
---|---|
SimVP-ConvMixer |
SimVP-ConvNeXt |
---|---|
SimVP-HorNet |
SimVP-MLPMixer |
---|---|
SimVP-MogaNet |
SimVP-Poolformer |
---|---|
SimVP-Swin |
SimVP-Uniformer |
---|---|
SimVP-VAN |
SimVP-ViT |
---|---|
Visualization of STL Benchmarks on Humidity (r)¶
Similar to Moving MNIST Benchmarks, we benchmark STL methods training 50 epochs with single GPU on r
(%). We provide config files in configs/weather/r_5_625 for 5.625
settings (\(32\times 64\) resolutions).
ConvLSTM |
---|
E3D-LSTM |
MAU |
---|---|
MIM |
PhyDNet |
---|---|
PredRNN |
PredRNN++ |
---|---|
PredRNN-V2 |
SimVP-V1 |
---|---|
SimVP-V2 |
TAU |
---|---|
SimVP-ConvMixer |
SimVP-ConvNeXt |
---|---|
SimVP-HorNet |
SimVP-MLPMixer |
---|---|
SimVP-MogaNet |
SimVP-Poolformer |
---|---|
SimVP-Swin |
SimVP-Uniformer |
---|---|
SimVP-VAN |
SimVP-ViT |
---|---|
Visualization of STL Benchmarks on Wind Component (uv10)¶
Similar to Moving MNIST Benchmarks, we benchmark popular STL methods training 50 epochs with single GPU on uv10
(ms-1). We provide config files in configs/weather/uv10_5_625 for 5.625
settings (\(32\times 64\) resolutions). Notice that the input data of uv10
has two channels.
ConvLSTM (in flow) |
ConvLSTM (out flow) |
---|---|
E3D-LSTM |
E3D-LSTM (out flow) |
---|---|
MAU (in flow) |
MAU (out flow) |
---|---|
MIM (in flow) |
MIM (out flow) |
---|---|
PhyDNet (in flow) |
PhyDNet (out flow) |
---|---|
PredRNN (in flow) |
PredRNN (out flow) |
---|---|
PredRNN++ (in flow) |
PredRNN++ (out flow) |
---|---|
PredRNN-V2 (in flow) |
PredRNN-V2 (out flow) |
---|---|
SimVP-V1 (in flow) |
SimVP-V1 (out flow) |
---|---|
SimVP-V2 (in flow) |
SimVP-V2 (out flow) |
---|---|
TAU (in flow) |
TAU (out flow) |
---|---|
SimVP-ConvMixer (in flow) |
SimVP-ConvMixer (out flow) |
---|---|
SimVP-ConvNeXt (in flow) |
SimVP-ConvNeXt (out flow) |
---|---|
SimVP-HorNet (in flow) |
SimVP-HorNet (out flow) |
---|---|
SimVP-MLPMixer (in flow) |
SimVP-MLPMixer (out flow) |
---|---|
SimVP-MogaNet (in flow) |
SimVP-MogaNet (out flow) |
---|---|
SimVP-Poolformer (in flow) |
SimVP-Poolformer (out flow) |
---|---|
SimVP-Uniformer (in flow) |
SimVP-Uniformer (out flow) |
---|---|
SimVP-VAN (in flow) |
SimVP-VAN (out flow) |
---|---|
SimVP-ViT (in flow) |
SimVP-ViT (out flow) |
---|---|
Visualization of STL Benchmarks on Cloud Cover (tcc)¶
Similar to Moving MNIST Benchmarks, we benchmark popular STL methods training 50 epochs with single GPU on tcc
(%). We provide config files in configs/weather/tcc_5_625 for 5.625
settings (\(32\times 64\) resolutions).
ConvLSTM |
---|
E3D-LSTM |
MAU |
---|---|
MIM |
PhyDNet |
---|---|
PredRNN |
PredRNN++ |
---|---|
PredRNN-V2 |
SimVP-V1 |
---|---|
SimVP-V2 |
TAU |
---|---|
SimVP-ConvMixer |
SimVP-ConvNeXt |
---|---|
SimVP-HorNet |
SimVP-MLPMixer |
---|---|
SimVP-MogaNet |
SimVP-Poolformer |
---|---|
SimVP-Swin |
SimVP-Uniformer |
---|---|
SimVP-VAN |
SimVP-ViT |
---|---|