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 and add small video test #64

Closed dHannasch closed 4 years ago

dHannasch commented 4 years ago

This packages PredNet for installation (using a cookiecutter) and adds a test using pytest that trains on a small video (from Hubble) included as a test resource.

The KITTI example is now in the external tests/ directory and imports prednet. Other applications would import prednet the same way.

This resolves #61.

APRashedAhmed commented 4 years ago

Any chance this will be merged in the near future? Seems to really improve upon the base repo.

dHannasch commented 4 years ago

Doesn't look like it. Closing this for now.