.. image:: https://github.com/kornia/data/raw/main/kornia_banner_pixie.png :align: center State-of-the-art and curated Computer Vision algorithms for AI. Kornia AI is on the mission to leverage and democratize the next generation of Computer Vision tools and Deep Learning libraries within the context of an Open Source community. .. code:: python >>> import kornia.geometry as K >>> registrator = K.ImageRegistrator('similarity') >>> model = registrator.register(img1, img2) Ready to use with state-of-the art Deep Learning models: DexiNed edge detection model. .. code-block:: python image = kornia.utils.sample.get_sample_images()[0][None] model = DexiNedBuilder.build() model.save(image) RTDETRDetector for object detection. .. code-block:: python image = kornia.utils.sample.get_sample_images()[0][None] model = RTDETRDetectorBuilder.build() model.save(image) BoxMotTracker for object tracking. .. code-block:: python import kornia image = kornia.utils.sample.get_sample_images()[0][None] model = BoxMotTracker() for i in range(4): model.update(image) model.save(image) Vision Transformer for image classification. .. code:: python >>> import torch.nn as nn >>> import kornia.contrib as K >>> classifier = nn.Sequential( ... K.VisionTransformer(image_size=224, patch_size=16), ... K.ClassificationHead(num_classes=1000), ... ) >>> logits = classifier(img) # BxN >>> scores = logits.argmax(-1) # B Multi-framework support ----------------------- You can now use Kornia with `NumPy `_, `TensorFlow `_, and `JAX `_. .. code:: python >>> import kornia >>> tf_kornia = kornia.to_tensorflow() .. raw:: html

Powered by

Join the community ------------------ - Join our social network communities with 1.8k+ members: - `Twitter `_: we share the recent research and news for out mainstream community. - `Slack `_: come to us and chat with our engineers and mentors to get support and resolve your questions. - Subscribe to our `YouTube channel `_ to get the latest video demos. ---- .. toctree:: :caption: GET STARTED :hidden: get-started/introduction get-started/highlights get-started/installation get-started/about Tutorials get-started/multi-framework-support get-started/training OpenCV AI Kit get-started/governance .. toctree:: :caption: API REFERENCE :maxdepth: 2 :hidden: augmentation color contrib core enhance feature filters geometry sensors io image losses models metrics morphology nerf onnx tracking testing utils x .. toctree:: :caption: KORNIA APPLICATIONS :hidden: applications/intro applications/visual_prompting applications/face_detection applications/image_augmentations applications/image_classification applications/image_matching applications/image_stitching applications/image_registration applications/image_denoising applications/semantic_segmentation applications/object_detection .. toctree:: :caption: KORNIA MODELS :hidden: models/efficient_vit models/rt_detr models/segment_anything models/mobile_sam models/yunet models/vit models/vit_mobile models/tiny_vit models/loftr models/defmo models/hardnet models/affnet models/sold2 models/dexined .. toctree:: :caption: SUPPORT :hidden: Issue tracker Slack community LibreCV community Twitter @kornia_foss community/chinese Kornia Youtube Kornia LinkedIn Kornia AI .. toctree:: :caption: COMMUNITY :hidden: community/contribute community/faqs community/bibliography