This is a Pytorch implementation of STConvLSTM, a spatiotemporal network for radar echo extrapolation in the following paper:
Spatiotemporal Convolutional LSTM for Radar Echo Extrapolation, by Shuxin Zhong, Xianxin Zeng, Qing Ling, Qiushi Wen, Weiguang Meng and Yerong Feng.
Required python libraries: Pytorch (>=1.3) + opencv + numpy
We conduct experiments on two datasets: Moving-MNIST and regional radar echo dataset.
Use the run.py script to train the model.
python run.py