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,638Updated last week
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆15,524Updated last year
• google-research / vision_transformer
  ☆12,119Updated 9 months ago
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆14,955Updated last year
• facebookresearch / ConvNeXt
  Code release for ConvNeXt model
  ☆6,235Updated 2 years ago
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆8,204Updated 3 years ago
• open-mmlab / mmdetection
  OpenMMLab Detection Toolbox and Benchmark
  ☆32,163Updated last year
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆6,337Updated last week
• jacobgil / pytorch-grad-cam
  Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…
  ☆12,451Updated 8 months ago
• kornia / kornia
  🐍 Geometric Computer Vision Library for Spatial AI
  ☆10,919Updated last week
• pytorch / vision
  Datasets, Transforms and Models specific to Computer Vision
  ☆17,373Updated this week
• facebookresearch / dino
  PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
  ☆7,348Updated last year
• open-mmlab / mmsegmentation
  OpenMMLab Semantic Segmentation Toolbox and Benchmark.
  ☆9,483Updated last year
• xmu-xiaoma666 / External-Attention-pytorch
  🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.…
  ☆12,135Updated last year
• qubvel-org / segmentation_models.pytorch
  Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
  ☆11,157Updated 2 weeks ago
• facebookresearch / deit
  Official DeiT repository
  ☆4,302Updated last year
• facebookresearch / mae
  PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
  ☆8,138Updated last year
• huawei-noah / Efficient-AI-Backbones
  Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
  ☆4,356Updated 9 months ago
• open-mmlab / mmpretrain
  OpenMMLab Pre-training Toolbox and Benchmark
  ☆3,798Updated 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,228Updated 5 months ago
• Cadene / pretrained-models.pytorch
  Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.
  ☆9,116Updated 3 years ago
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,873Updated this week
• facebookresearch / SlowFast
  PySlowFast: video understanding codebase from FAIR for reproducing state-of-the-art video models.
  ☆7,219Updated last month
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆9,312Updated 3 weeks ago
• facebookresearch / moco
  PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
  ☆5,103Updated 2 months ago
• Lyken17 / pytorch-OpCounter
  Count the MACs / FLOPs of your PyTorch model.
  ☆5,063Updated last year
• KevinMusgrave / pytorch-metric-learning
  The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
  ☆6,275Updated 4 months ago
• mlfoundations / open_clip
  An open source implementation of CLIP.
  ☆13,113Updated last month
• cmhungsteve / Awesome-Transformer-Attention
  An ultimately comprehensive paper list of Vision Transformer/Attention, including papers, codes, and related websites
  ☆4,979Updated last year
• utkuozbulak / pytorch-cnn-visualizations
  Pytorch implementation of convolutional neural network visualization techniques
  ☆8,176Updated 11 months ago