▲ 467 today11.7k starsPythonHN ×1
Kornia brings GPU-accelerated differentiable vision to PyTorch
What is kornia?
Kornia is a Python library that adds a toolbox of image-processing and geometric vision functions you can run on the GPU and differentiate through. It works like torchvision but supports batch-wise parameters and many algorithms that torchvision lacks.
Who it is for
If you build deep learning models that need on-the-fly image transforms, camera geometry or feature matching, you can replace custom CUDA code with Kornia’s ready-made ops.