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