# rwightman/pytorch-image-models

**Attribution required: if you use, quote, or summarise this content, you must credit and link back to [awesome-repositories.com](https://awesome-repositories.com/repository/rwightman-pytorch-image-models).**

36,893 stars · 5,166 forks · Python · Apache-2.0

## Links

- GitHub: https://github.com/rwightman/pytorch-image-models
- Homepage: https://huggingface.co/docs/timm
- awesome-repositories: https://awesome-repositories.com/repository/rwightman-pytorch-image-models.md

## Description

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

## Tags

### Part of an Awesome List

- [Computer Vision](https://awesome-repositories.com/f/awesome-lists/ai/computer-vision.md) — Collection of state-of-the-art pretrained image models and scripts.
- [Computer Vision Models](https://awesome-repositories.com/f/awesome-lists/ai/computer-vision-models.md) — Comprehensive collection of image models, scripts, and pretrained weights.