Starred repositories
A paper list of object detection using deep learning.
Easy usage of distillation tokens with vision transformers. Includes models pretrained on ImageNet.
Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch
Keras implementation of ViT (Vision Transformer)
Keras Implementation of Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
Copy-paste augmentation for segmentation and detection tasks
YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
PyTorch3D is FAIR's library of reusable components for deep learning with 3D data
A modular framework for vision & language multimodal research from Facebook AI Research (FAIR)
End-to-End Object Detection with Transformers
Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch
On the Variance of the Adaptive Learning Rate and Beyond
A collection of loss functions for medical image segmentation
A PyTorch implementation of EfficientNet
Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
a Ready-to-use PyTorch Extension of Unofficial CutMix Implementations with more improved performance.
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (V…
The official Pytorch implementation of OCNet, OCRNet, and SegFix.
Learning Dynamic Routing for Semantic Segmentation
Nvidia Semantic Segmentation monorepo
Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, …
The fastai book, published as Jupyter Notebooks
Free online textbook of Jupyter notebooks for fast.ai Computational Linear Algebra course