PredNet_pytorch
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
