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

GitHub
32k stars
311 watching
5k forks
Language: Python
last commit: 6 days ago
Linked from 4 awesome lists

augmixconvnextdistributed-trainingdual-path-networksefficientnetimage-classificationimagenetmaxvitmixnetmobile-deep-learningmobilenet-v2mobilenetv3nfnetsnormalization-free-trainingpretrained-modelspretrained-weightspytorchrandaugmentresnetvision-transformer-models

Screenshot of huggingface/pytorch-image-models website

https://huggingface.co/docs/timm

Backlinks from these awesome lists: