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…
  ☆24,941Updated last week
• google-research / vision_transformer
  ☆12,260Updated 2 weeks ago
• jacobgil / pytorch-grad-cam
  Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…
  ☆12,571Updated 9 months ago
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆15,682Updated last year
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆8,213Updated 3 years ago
• facebookresearch / ConvNeXt
  Code release for ConvNeXt model
  ☆6,280Updated 3 years ago
• albumentations-team / albumentations
  Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
  ☆15,266Updated 7 months ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆15,071Updated last year
• facebookresearch / mae
  PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
  ☆8,204Updated last year
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆6,388Updated 2 weeks ago
• facebookresearch / dino
  PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
  ☆7,424Updated last year
• utkuozbulak / pytorch-cnn-visualizations
  Pytorch implementation of convolutional neural network visualization techniques
  ☆8,195Updated last year
• facebookresearch / deit
  Official DeiT repository
  ☆4,318Updated last year
• eriklindernoren / PyTorch-GAN
  PyTorch implementations of Generative Adversarial Networks.
  ☆17,415Updated last year
• facebookresearch / moco
  PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
  ☆5,109Updated 2 weeks ago
• Cadene / pretrained-models.pytorch
  Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
  ☆9,115Updated 3 years ago
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,899Updated last week
• kornia / kornia
  🐍 Geometric Computer Vision Library for Spatial AI
  ☆11,031Updated last week
• qubvel-org / segmentation_models.pytorch
  Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
  ☆11,296Updated last month
• KevinMusgrave / pytorch-metric-learning
  The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
  ☆6,300Updated 5 months ago
• Lightning-AI / pytorch-lightning
  Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
  ☆30,779Updated this week
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆9,370Updated this week
• facebookresearch / SlowFast
  PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
  ☆7,265Updated last week
• open-mmlab / mmsegmentation
  OpenMMLab Semantic Segmentation Toolbox and Benchmark.
  ☆9,584Updated last year
• dk-liang / Awesome-Visual-Transformer
  Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)
  ☆3,567Updated last year
• open-mmlab / mmpretrain
  OpenMMLab Pre-training Toolbox and Benchmark
  ☆3,811Updated last year
• open-mmlab / mmdetection
  OpenMMLab Detection Toolbox and Benchmark
  ☆32,319Updated last year
• lanpa / tensorboardX
  tensorboard for pytorch (and chainer, mxnet, numpy, ...)
  ☆7,987Updated 2 months ago
• cmhungsteve / Awesome-Transformer-Attention
  An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites
  ☆5,004Updated last year
• open-mmlab / mmselfsup
  OpenMMLab Self-Supervised Learning Toolbox and Benchmark
  ☆3,298Updated 2 years ago