kornia

github.com ↗

Open Source Differentiable Computer Vision Library for PyTorch.