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.
>>> import kornia.geometry as K >>> registrator = K.ImageRegistrator('similarity') >>> model = registrator(img1, img2)
Ready to use with state-of-the art Deep Learning models:
>>> 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
Join the community#
Subscribe to our YouTube channel to get the latest video demos.