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,907Updated last week
• google-research / vision_transformer
  ☆12,239Updated last week
• 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 / ConvNeXt
  Code release for ConvNeXt model
  ☆6,280Updated 3 years ago
• lukemelas / EfficientNet-PyTorch
  A PyTorch implementation of EfficientNet
  ☆8,213Updated 3 years ago
• microsoft / Swin-Transformer
  This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".
  ☆15,665Updated last year
• facebookresearch / detr
  End-to-End Object Detection with Transformers
  ☆15,071Updated last year
• 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
• facebookresearch / deit
  Official DeiT repository
  ☆4,318Updated last year
• open-mmlab / mmcv
  OpenMMLab Computer Vision Foundation
  ☆6,388Updated last week
• utkuozbulak / pytorch-cnn-visualizations
  Pytorch implementation of convolutional neural network visualization techniques
  ☆8,195Updated last year
• facebookresearch / mae
  PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
  ☆8,194Updated last year
• facebookresearch / dino
  PyTorch code for Vision Transformers training with the Self-Supervised learning method DINO
  ☆7,412Updated last year
• qubvel-org / segmentation_models.pytorch
  Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.
  ☆11,283Updated last month
• NVIDIA / apex
  A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
  ☆8,899Updated this week
• open-mmlab / mmsegmentation
  OpenMMLab Semantic Segmentation Toolbox and Benchmark.
  ☆9,584Updated last year
• arogozhnikov / einops
  Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)
  ☆9,358Updated 2 weeks ago
• pytorch / vision
  Datasets, Transforms and Models specific to Computer Vision
  ☆17,463Updated this week
• open-mmlab / mmpretrain
  OpenMMLab Pre-training Toolbox and Benchmark
  ☆3,811Updated last year
• Lyken17 / pytorch-OpCounter
  Count the MACs / FLOPs of your PyTorch model.
  ☆5,075Updated last year
• kornia / kornia
  🐍 Geometric Computer Vision Library for Spatial AI
  ☆11,031Updated last week
• dk-liang / Awesome-Visual-Transformer
  Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)
  ☆3,562Updated last year
• open-mmlab / mmdetection
  OpenMMLab Detection Toolbox and Benchmark
  ☆32,319Updated last year
• vdumoulin / conv_arithmetic
  A technical report on convolution arithmetic in the context of deep learning
  ☆14,591Updated 2 years ago
• aleju / imgaug
  Image augmentation for machine learning experiments.
  ☆14,722Updated 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,156Updated last year
• huawei-noah / Efficient-AI-Backbones
  Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
  ☆4,375Updated 10 months ago
• facebookresearch / moco
  PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
  ☆5,109Updated 2 weeks ago
• amusi / CVPR2025-Papers-with-Code
  CVPR 2025 论文和开源项目合集
  ☆21,813Updated 6 months ago
• KevinMusgrave / pytorch-metric-learning
  The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
  ☆6,297Updated 5 months ago