Giter Club home page Giter Club logo

ms-rnn's Introduction

Source Code for Papers:

MS-RNN: A Flexible Multi-Scale Framework for Spatiotemporal Predictive Learning

PrecipLSTM: A Meteorological Spatiotemporal LSTM for Precipitation Nowcasting

MS-LSTM: Exploring Spatiotemporal Multiscale Representations in Video Prediction Domain

Reproduced Models:

ConvRNNs MS-RNNs
ConvLSTM MS-ConvLSTM
TrajGRU MS-TrajGRU
PredRNN MS-PredRNN
PredRNN++ MS-PredRNN++
MIM MS-MIM
MotionRNN MS-MotionRNN
PredRNN-V2 MS-PredRNN-V2
PrecipLSTM MS-PrecipLSTM
CMS-LSTM MS-CMS-LSTM
MoDeRNN MS-MoDeRNN
MK-LSTM MS-LSTM

Installing Libraries:

* Installing Libraries:

pip3 install -r requirements.txt

* Installing CUDA (Only Needed for Reproducing PrecipLSTM/MS-PrecipLSTM):

Higher versions of CUDA are not supported, and CUDA 11.1 is recommended. https://blog.csdn.net/qq_40947610/article/details/114757551

* Installing Local Attention (Only Needed for Reproducing PrecipLSTM/MS-PrecipLSTM):

cd img_local_att
python setup.py install

https://github.com/zzd1992/Image-Local-Attention

Hyperparameters:

See connfig.py

Running:

python -m torch.distributed.launch --nproc_per_node=4 main.py

Citation:

If you find this repository useful, please cite the following papers.
@article{ma2022ms,
  title={MS-RNN: A flexible multi-scale framework for spatiotemporal predictive learning},
  author={Ma, Zhifeng and Zhang, Hao and Liu, Jie},
  journal={arXiv preprint arXiv:2206.03010},
  year={2022}
}
@article{ma2022preciplstm,
  title={PrecipLSTM: A Meteorological Spatiotemporal LSTM for Precipitation Nowcasting},
  author={Ma, Zhifeng and Zhang, Hao and Liu, Jie},
  journal={IEEE Transactions on Geoscience and Remote Sensing},
  volume={60},
  pages={1--8},
  year={2022},
  publisher={IEEE}
}
@article{ma2023ms,
  title={MS-LSTM: Exploring Spatiotemporal Multiscale Representations in Video Prediction Domain},
  author={Ma, Zhifeng and Zhang, Hao and Liu, Jie},
  journal={arXiv preprint arXiv:2304.07724},
  year={2023}
}

ms-rnn's People

Contributors

mazhf avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.