kornia stable
PyTorch · differentiable computer vision
The research bench: geometry, features and augmentations written to be differentiated, inside the training loop.
- Image filters, colour, geometry and morphology that gradients flow through
- Augmentation pipelines that run on the GPU, batched and reproducible
- Camera models, epipolar geometry, homographies and 3D transforms
- Feature detection, description and matching, from HardNet to LightGlue
- Exports to ONNX and compiles with torch.compile; try it in the playground
pip install kornia