kornia / kornia
π Geometric Computer Vision Library for Spatial AI
β10,461Updated this week
Alternatives and similar repositories for kornia
Users that are interested in kornia are comparing it to the libraries listed below
Sorting:
- Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125β14,931Updated this week
- End-to-End Object Detection with Transformersβ14,331Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)β8,903Updated 3 weeks ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --β¦β34,105Updated this week
- PyTorch3D is FAIR's library of reusable components for deep learning with 3D dataβ9,239Updated last month
- Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.β31,948Updated 3 weeks ago
- Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pyβ¦β22,812Updated 2 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorchβ8,661Updated this week
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.β3,283Updated last year
- Code release for ConvNeXt modelβ5,996Updated 2 years ago
- β11,336Updated 2 months ago
- Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.β29,439Updated last week
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".β14,755Updated 9 months ago
- Official DeiT repositoryβ4,195Updated last year
- PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.β6,934Updated 5 months ago
- Datasets, Transforms and Models specific to Computer Visionβ16,824Updated this week
- A PyTorch implementation of EfficientNetβ8,107Updated 3 years ago
- π A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (iβ¦β8,708Updated this week
- Google Brain AutoMLβ6,366Updated 2 months ago
- PyTorch code for Vision Transformers training with the Self-Supervised learning method DINOβ6,831Updated 10 months ago
- OpenMMLab Computer Vision Foundationβ6,124Updated 3 weeks ago
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)β7,939Updated 2 weeks ago
- A python library for self-supervised learning on images.β3,379Updated last week
- Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.β10,430Updated this week
- Image augmentation for machine learning experiments.β14,573Updated 9 months ago
- A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learβ¦β5,401Updated this week
- Model interpretability and understanding for PyTorchβ5,209Updated last week
- PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722β4,973Updated 2 weeks ago
- Collection of common code that's shared among different research projects in FAIR computer vision team.β2,126Updated 5 months ago
- Model summary in PyTorch similar to `model.summary()` in Kerasβ4,040Updated last year