PredNet_pytorch icon indicating copy to clipboard operation
PredNet_pytorch copied to clipboard

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

trafficstars

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

example