A rookie modified ConvLSTM source code.
Rookies in deep learning who mistakenly plan to use ConvLSTM for final year project (just like me) or something. Anyone familiared with CNN and LSTM will find this repo not worthy of reading.
This repo provide a shortpath on how to use the current source code on your dataset. There are some redundant codes.
Majority of code comes from https://github.com/jhhuang96/ConvLSTM-PyTorch.
https://github.com/holmdk/Video-Prediction-using-PyTorch also deploy ConvLSTM by Pytorch-lighting. The codes of models are much more concise and understandable for beginners.
You need to modify the Decoder and Encoder's parameter, depends on your original matrix widths and lengths. You need to do some calculation to decide the size of each CNN layers's kernel. Thats all you need to input the data.
I think the hard part of starting is to figure out what kind and format of the data you need to input.
Rainnet: in ndf format:
Tianchi 2017
HKO(student need their advisor to send application)
Tensorboard only add_image the first epoch of image in one run.