Traffic Prediction Visualization¶
We provide visualization of spatiotemporal prediction learning (STL) methods on popular traffic prediction datasets. 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)
Visualization of TaxiBJ Benchmarks¶
We provide visualization figures of various traffic prediction methods on various benchmarks. 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 the first channel of TaxiBJ with the script:
python tools/visualizations/vis_video.py -d taxibj -w work_dirs/exp_name --vis_channel 0 --index 0 --save_dirs fig_taxibj_vis
We provide GIF visualizations of experiments in configs/taxibj for TaxiBJ (32x32 resolutions).
ConvLSTM (in flow) |
ConvLSTM (out flow) |
---|---|
DMVFN (in flow) |
DMVFN (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) |
---|---|