Skip to content
master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
src
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

torch-submod

Documentation Status Build Status

A PyTorch library for differentiable submodular minimization.

At the moment only one- and two-dimensional graph cut functions have been implemented, so that this package provides differentiable (with respect to the input signal and the weights) total variation solvers.

Please refer to the documentation for more information about the project. You can also have a look at the following notebook that showcases how to learn weights for image denoising.

Installation

After installing PyTorch, you can install the package with:

python setup.py install

Testing

To run the tests you simply have to run:

python setup.py test

Bibliography

  • [DK17] J. Djolonga and A. Krause. Differentiable learning of submodular models. In Advances in Neural Information Processing Systems (NIPS), 2017.
  • [NB17] V. Niculae and M. Blondel. A regularized framework for sparse and structured neural attention. arXiv preprint arXiv:1705.07704, 2017.

About

A PyTorch library for differentiable submodular minimization.

Resources

License

Releases

No releases published

Packages

No packages published