thuml / predrnn-pytorch
Official implementation for NIPS'17 paper: PredRNN: Recurrent Neural Networks for Predictive Learning Using Spatiotemporal LSTMs.
☆454Updated last year
Alternatives and similar repositories for predrnn-pytorch:
Users that are interested in predrnn-pytorch are comparing it to the libraries listed below
- ConvLSTM/ConvGRU (Encoder-Decoder) with PyTorch on Moving-MNIST☆405Updated 2 years ago
- pytorch implemention of trajGRU.☆570Updated 5 years ago
- Code release for "PredRNN++: Towards A Resolution of the Deep-in-Time Dilemma in Spatiotemporal Predictive Learning" (ICML 2018)☆264Updated 5 years ago
- Video Predicting using ConvLSTM and pytorch☆114Updated 3 years ago
- Code release for "Memory In Memory: A Predictive Neural Network for Learning Higher-Order Non-Stationarity from Spatiotemporal Dynamics" …☆153Updated 5 years ago
- About Code release for "MotionRNN: A Flexible Model for Video Prediction with Spacetime-Varying Motions" (CVPR 2021) https://arxiv.org/ab…☆53Updated last year
- ☆90Updated 6 months ago
- Self-Attention ConvLSTM for Spatiotemporal Prediction, described in `https://ojs.aaai.org//index.php/AAAI/article/view/6819`, test on Mov…☆63Updated 3 years ago
- Pytorch implementation of Self-Attention ConvLSTM☆81Updated 6 months ago
- Implementation of ConvLSTM in pytorch applied for BCI (Brain Machine Interface) following paper: Convolutional LSTM Network-A Machine Lea…☆101Updated 4 years ago
- Source code of paper "[NIPS2017] Deep Learning for Precipitation Nowcasting: A Benchmark and A New Model"☆403Updated 2 years ago
- Code for our CVPR 2020 paper "Disentangling Physical Dynamics from Unknown Factors for UnsupervisedVideo Prediction"☆187Updated 2 years ago
- Pytorch implementations of ConvLSTM and ConvGRU modules with examples☆74Updated 3 years ago
- The official implementation of the CVPR'22 paper SimVP: Simpler Yet Better Video Prediction.☆231Updated last year
- This is a Pytorch implementation of PredRNN++☆38Updated 4 years ago
- TianChi AIEarth Contest Solution☆80Updated 3 years ago
- 从Convlstm2D的基础上改编: https://github.com/ndrplz/ConvLSTM_pytorch 可用于雷达回波等短临预报领域,采用模仿Dr.SXJ的Seq2seq结构.☆33Updated 4 years ago
- PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"☆242Updated 8 months ago
- 时空预测模型PyTorch复现☆51Updated 2 years ago
- e3d-lstm; Eidetic 3D LSTM A Model for Video Prediction and Beyond☆199Updated 2 years ago
- Moving MNIST as PyTorch Dataset☆96Updated 4 years ago
- Convolutional LSTM for Precipitation Nowcasting☆55Updated 6 years ago
- PredRNN: Recurrent Neural Networks for Predictive Learning using Spatiotemporal LSTMs☆124Updated 5 years ago
- Using the Pytorch to build an image temporal prediction model of the encoder-forecaster structure, ConvGRU kernel & ConvLSTM kernel☆112Updated 5 years ago
- Implementation of TAAConvLSTM and SAAConvLSTM used in "Attention Augmented ConvLSTM for Environment Prediction"☆55Updated 3 years ago
- Pytorch Implementation of the Paper: Self-Attention ConvLSTM for Spatiotemporal Prediction☆15Updated 3 months ago
- STRPM: A Spatiotemporal Residual Predictive Model for High-Resolution Video Prediction, CVPR2022☆17Updated 2 years ago
- CIKM contest entry 'Convolutional LSTM neural network to extrapolate radar images, and predict rainfall'☆191Updated 7 years ago
- Code for the paper "STConvS2S: Spatiotemporal Convolutional Sequence to Sequence Network for Weather Forecasting" (Neurocomputing, Elsevi…☆65Updated 3 years ago
- Implementation of DeepMind's Deep Generative Model of Radar (DGMR) https://arxiv.org/abs/2104.00954☆225Updated this week