Differentiable Computer Vision for PyTorch — Kornia

kornia.readthedocs.io

Kornia is the open-source differentiable computer vision library for PyTorch, with GPU-accelerated image processing, geometry, augmentation and vision models.