Home

Awesome

Lightweight Probabilistic Deep Networks

This source code accompanies the paper:<br>

Lightweight Probabilistic Deep Networks, Jochen Gast and Stefan Roth, CVPR 2018.

Framework Requirements

This code has been built and tested with PyTorch 0.4.0 on Python 2.7. Please check PyTorch for requirements and setup instructions. Typically, the easiest way is to install a full Anaconda Python distribution.

Custom packages

After having installed Anaconda, following custom packages are required: setproctitle tqdm. They can be installed via the one-liner

$ pip install setproctitle tqdm

Pretrained models

ModelCheckpoint
FlowNetADFLink
FlowNetProbOutLink

Contact

If you have further questions, feel free to contact jochen.gast@visinf.tu-darmstadt.de .

Citation

If you use our code, please cite our CVPR 2018 paper:

@inproceedings{Gast:2018:LPD,
  title={Lightweight Probabilistic Deep Networks},
  author={Jochen Gast and Stefan Roth},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year={2018}
}