Alternatives and similar repositories for pytorch-image-models

Users that are interested in pytorch-image-models are comparing it to the libraries listed below

• lucidrains / vit-pytorch
  Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Py…
  ☆23,654Updated this week
• google-research / vision_transformer
  ☆11,675Updated 5 months ago
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆15,108Updated last year
• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆15,109Updated last month
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆8,147Updated 3 years ago
• facebookresearch / ConvNeXt
  Code release for ConvNeXt model
  ☆6,102Updated 2 years ago
• jacobgil / pytorch-grad-cam
  Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…
  ☆12,066Updated 4 months ago
• open-mmlab / mmdetection
  OpenMMLab Detection Toolbox and Benchmark
  ☆31,594Updated last year
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆6,223Updated 3 months ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆14,625Updated last year
• open-mmlab / mmsegmentation
  OpenMMLab Semantic Segmentation Toolbox and Benchmark.
  ☆9,166Updated last year
• qubvel-org / segmentation_models.pytorch
  Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
  ☆10,804Updated last week
• KevinMusgrave / pytorch-metric-learning
  The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
  ☆6,207Updated this week
• pytorch / vision
  Datasets, Transforms and Models specific to Computer Vision
  ☆17,082Updated this week
• facebookresearch / dino
  PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
  ☆7,088Updated last year
• facebookresearch / mae
  PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
  ☆7,949Updated last year
• facebookresearch / deit
  Official DeiT repository
  ☆4,250Updated last year
• kornia / kornia
  🐍 Geometric Computer Vision Library for Spatial AI
  ☆10,692Updated this week
• aleju / imgaug
  Image augmentation for machine learning experiments.
  ☆14,656Updated last year
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on multiple GPUs, TPUs with zero code changes.
  ☆29,980Updated last week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆9,127Updated last week
• Cadene / pretrained-models.pytorch
  Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
  ☆9,096Updated 3 years ago
• huawei-noah / Efficient-AI-Backbones
  Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
  ☆4,280Updated 5 months ago
• google / automl
  Google Brain AutoML
  ☆6,391Updated 5 months ago
• open-mmlab / mmpretrain
  OpenMMLab Pre-training Toolbox and Benchmark
  ☆3,734Updated 9 months ago
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,774Updated this week
• facebookresearch / moco
  PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
  ☆5,046Updated last month
• mlfoundations / open_clip
  An open source implementation of CLIP.
  ☆12,447Updated 2 weeks ago
• cmhungsteve / Awesome-Transformer-Attention
  An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites
  ☆4,918Updated last year
• dk-liang / Awesome-Visual-Transformer
  Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)
  ☆3,520Updated 7 months ago