search

coxlab / prednet

Code and models accompanying "Deep Predictive Coding Networks for Video Prediction and Unsupervised Learning"
https://arxiv.org/abs/1605.08104
MIT License
759 stars 259 forks source link

Package PredNet for installation #61

Closed dHannasch closed 3 years ago

dHannasch commented 4 years ago

Would you be interested in getting PredNet packaged up for installation via pip?

You can see how this might work at https://github.com/dHannasch/prednet/tree/use-cookiecutter.

The idea would be to allow people to install and use the core PredNet:

$ pip install git+ssh://git@github.com/dHannasch/prednet.git@use-cookiecutter
Successfully built prednet
$ python -c "import prednet"

Specific applications, such as to the KITTI data, would then import prednet. For example, kitti_train.py would include these lines:

from prednet.prednet import PredNet
from prednet.data_utils import SequenceGenerator