pytorch-image-models
by huggingface (organization)
The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet… (description from the project’s GitHub page)
Repository facts
- Stars
- 37,226
- Forks
- 5,216
- Open issues and pull requests
- 20
- Primary language
- Python
- License
- Apache License 2.0 (Apache-2.0)
- Owner
- huggingface · Organization
- Created
- Feb 2, 2019
- Last push
- Oct 17, 2026
Topics
Related projects
Projects that share this one’s GitHub topics.
Rank 9,194: calmiLovesAI/Basic_CNNs_TensorFlow2
A tensorflow2 implementation of some basic CNNs(MobileNetV1/V2/V3, EfficientNet, ResNeXt, InceptionV4, InceptionResNetV1/V2, SENet, SqueezeNet, DenseNet, ShuffleNetV2, ResNet).
532 starsPythonMITLast push Nov 24, 2021
Rank 8,285: alvinwan/neural-backed-decision-trees
Making decision trees competitive with neural networks on CIFAR10, CIFAR100, TinyImagenet200, Imagenet
625 starsPythonMITLast push May 1, 2023
Rank 2,049: open-mmlab/mmpretrain
OpenMMLab Pre-training Toolbox and Benchmark
3,860 starsPythonApache-2.0Last push Nov 1, 2024
Rank 3,515: qubvel/efficientnet
Implementation of EfficientNet model. Keras and TensorFlow Keras.
2,103 starsPythonApache-2.0Last push Jan 24, 2024
Rank 6,943: sail-sg/Adan
Adan: Adaptive Nesterov Momentum Algorithm for Faster Optimizing Deep Models
823 starsPythonApache-2.0Last push Jun 8, 2025
Rank 4,437: rwightman/gen-efficientnet-pytorch
Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS
1,577 starsPythonApache-2.0Last push Jun 13, 2024