pytorch-image-models
huggingface/pytorch-image-models
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, ConvNeXt, and more
36,375stars
Forks
5,116
Open issues
65
Watchers
36,375
Size
29.9 MB
PythonApache License 2.0
augmixconvnextdistributed-trainingefficientnetimage-classificationimagenetmaxvitmixnetmobile-deep-learningmobilenet-v2mobilenetv3nfnetsnormalization-free-trainingoptimizerpretrained-modelspretrained-weightspytorchrandaugmentresnetvision-transformer-models
Created: Feb 2, 2019
Updated: Feb 18, 2026
Last push: Feb 15, 2026