- PyTorch: 1.8
- Python: 3.7
- Other python packages: please refer to the following code
- Arguments.py: model configuration of all parameters
- convlstm.py: ConvLSTM model structure
- Convtrain.py: training and testing part of the code for hybrid model
- get_frames.py: Extract video frames
- Visual test metrics: plot.py, plot_tiles.py, plot_time.py, plot_vs.py