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

• TaxiBJ Benchmarks

Currently supported spatiotemporal prediction methods

• ConvLSTM (NeurIPS'2015)
• PredNet (ICLR'2017)
• PredRNN (NeurIPS'2017)
• PredRNN++ (ICML'2018)
• E3D-LSTM (ICLR'2018)
• MIM (CVPR'2019)
• CrevNet (ICLR'2020)
• PhyDNet (CVPR'2020)
• MAU (NeurIPS'2021)
• PredRNN.V2 (TPAMI'2022)
• SimVP (CVPR'2022)
• SimVP.V2 (ArXiv'2022)
• TAU (CVPR'2023)
• DMVFN (CVPR'2023)

Currently supported MetaFormer models for SimVP

• ViT (ICLR'2021)
• Swin-Transformer (ICCV'2021)
• MLP-Mixer (NeurIPS'2021)
• ConvMixer (Openreview'2021)
• UniFormer (ICLR'2022)
• PoolFormer (CVPR'2022)
• ConvNeXt (CVPR'2022)
• VAN (ArXiv'2022)
• IncepU (SimVP.V1) (CVPR'2022)
• gSTA (SimVP.V2) (ArXiv'2022)
• HorNet (NeurIPS'2022)
• 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)

(back to top)