kornia / kornia
🐍 Geometric Computer Vision Library for Spatial AI
☆10,183Updated this week
Alternatives and similar repositories for kornia:
Users that are interested in kornia are comparing it to the libraries listed below
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125☆14,544Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆33,062Updated this week
- End-to-End Object Detection with Transformers☆13,934Updated 10 months ago
- PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO☆6,557Updated 7 months ago
- PyTorch3D is FAIR's library of reusable components for deep learning with 3D data☆9,020Updated last week
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,079Updated last month
- Code release for ConvNeXt model☆5,861Updated 2 years ago
- A PyTorch implementation of EfficientNet☆8,012Updated 2 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,697Updated this week
- ☆10,848Updated 2 months ago
- Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.☆10,003Updated last week
- Image augmentation for machine learning experiments.☆14,501Updated 6 months ago
- OpenMMLab Semantic Segmentation Toolbox and Benchmark.☆8,518Updated 5 months ago
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.☆28,905Updated this week
- Official DeiT repository☆4,125Updated 10 months ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆14,256Updated 6 months ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,052Updated 2 years ago
- PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722☆4,868Updated last month
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,265Updated 11 months ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,907Updated 3 weeks ago
- SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners☆4,177Updated last year
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,523Updated last month
- High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.☆4,578Updated 3 weeks ago
- PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.☆6,771Updated 2 months ago
- Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.☆31,151Updated 3 weeks ago
- OpenMMLab Computer Vision Foundation☆5,992Updated 2 weeks ago
- Sandbox for training deep learning networks☆2,991Updated 5 months ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…☆21,634Updated 2 weeks ago
- OpenMMLab Detection Toolbox and Benchmark☆30,140Updated 5 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,027Updated 11 months ago