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.

sh train.sh

sh evaluate.sh

Name		Name	Last commit message	Last commit date
debug		debug
kitti_results/prediction_plots		kitti_results/prediction_plots
model_data_keras2		model_data_keras2
.gitattributes		.gitattributes
LICENSE		LICENSE
README.md		README.md
data_utils.py		data_utils.py
debug.py		debug.py
evaluate.py		evaluate.py
evaluate.sh		evaluate.sh
load_weights.py		load_weights.py
prednet.py		prednet.py
train.py		train.py
train.sh		train.sh
visualization.py		visualization.py