pytorch-image-models
by huggingface
Pythonpushed almost 2 years ago
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
AI summary
Image classification models
A collection of pre-trained PyTorch models and optimizers for image classification tasks
- stars
- 32.6K
- forks
- 4.8K
- watching
- 313
- awesome lists
- 4
Featured in 4 awesome lists
Each link jumps to the spot where the list mentions pytorch-image-models.