Skip to content
/ pytorch-image-models Public

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

huggingface.co/docs/timm

License

Apache-2.0 license
34.4k stars 4.9k forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

huggingface/pytorch-image-models

Error
Looks like something went wrong!

About

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

huggingface.co/docs/timm

Topics

optimizer pytorch imagenet image-classification resnet pretrained-models mixnet pretrained-weights distributed-training mobilenet-v2 mobile-deep-learning mobilenetv3 efficientnet augmix randaugment nfnets normalization-free-training vision-transformer-models convnext maxvit

Resources

Readme

License

Apache-2.0 license

Code of conduct

Code of conduct
Activity
Custom properties

Stars

34.4k stars

Watchers

316 watching

Forks

4.9k forks
Report repository

Releases 65

Release v1.0.15 Latest
Feb 23, 2025
+ 64 releases

Sponsor this project

Packages

No packages published

Contributors 150
+ 136 contributors

Languages