xingyu-liu / PredNet_pytorch

A pytorch reimplement of PredNet, which is a deep recurrent convolutional neural network for video prediction and unsupervised learning.

PredNet_pytorch

An implement of PredNet in pytorch. See the paper Deep predictive coding networks for video prediction and unsupervised learning in ICLR 2017 for more details.

The offical code is implemented via Keras, and the project website can be found at https://coxlab.github.io/prednet/.

Dataset

The preprocessed KITTI data can be obtained using downlaod_data.sh in offical code.

How to run

Train model

sh train.sh

Evaluate model

sh evaluate.sh

Some example results