A differentiable computer vision library for PyTorch, providing geometric vision and image processing algorithms for AI workflows.
Kornia is a differentiable computer vision library built on PyTorch that provides a rich set of image processing and geometric vision algorithms. It enables seamless integration of classic computer vision techniques into deep learning pipelines, supporting tasks like image transformations, augmentations, and AI-driven vision models. The library focuses on making these operations differentiable and GPU-accelerated for modern AI workflows.
Researchers and developers working on computer vision and deep learning projects, particularly those using PyTorch who need differentiable vision operators, data augmentation pipelines, or pre-trained vision models.
Developers choose Kornia for its deep integration with PyTorch, providing a comprehensive suite of differentiable vision operations that are not available in standard deep learning frameworks. Its unique selling point is bringing traditional computer vision algorithms into the differentiable programming paradigm, enabling end-to-end trainable vision systems.
🐍 Geometric Computer Vision Library for Spatial AI
Open-Awesome is built by the community, for the community. Submit a project, suggest an awesome list, or help improve the catalog on GitHub.
Kornia is built directly on PyTorch, enabling seamless use of automatic differentiation and GPU acceleration for batch vision operations, as highlighted in its seamless AI workflow integration.
Offers over 500 differentiable image processing and vision algorithms, including filters, transformations, and advanced augmentations like AutoAugment, making it a one-stop shop for trainable vision systems.
Through the Ivy framework, Kornia extends support to TensorFlow, JAX, and NumPy, allowing use beyond PyTorch ecosystems, as mentioned in the multi-framework support section.
Provides extensive float16 and bfloat16 support across many modules for optimized computation on modern hardware, detailed in the half-precision table with test results showing high pass rates.
Many modules have only partial or no support for half-precision types, leading to potential inaccuracies or failures in precision-sensitive operations like calibration, as admitted in the README's precision guide.
Recommends using Pixi for environment management in development, which adds complexity over standard pip installations and may deter quick prototyping or contributions.
The AI policy requires contributors to be sole authors with proof of verification, discouraging community-driven enhancements and potentially slowing down bug fixes or new feature adoption.