https://github.com/rwightman/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
https://github.com/rwightman/pytorch-image-models

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

Last synced: 3 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/rwightman/pytorch-image-models
• Owner: huggingface
• License: apache-2.0
• Created: 2019-02-02T05:51:12.000Z (about 6 years ago)
• Default Branch: main
• Last Pushed: 2025-01-19T23:05:30.000Z (3 months ago)
• Last Synced: 2025-01-20T13:03:20.448Z (3 months ago)
• Topics: augmix, convnext, distributed-training, dual-path-networks, efficientnet, image-classification, imagenet, maxvit, mixnet, mobile-deep-learning, mobilenet-v2, mobilenetv3, nfnets, normalization-free-training, pretrained-models, pretrained-weights, pytorch, randaugment, resnet, vision-transformer-models
• Language: Python
• Homepage: https://huggingface.co/docs/timm
• Size: 26.5 MB
• Stars: 32,934
• Watchers: 317
• Forks: 4,812
• Open Issues: 67
• Metadata Files:
  • Readme: README.md
  • Contributing: CONTRIBUTING.md
  • Funding: .github/FUNDING.yml
  • License: LICENSE

Awesome Lists containing this project

• Algorithms-Cheatsheet-Resources - PyTorch image models, scripts, pretrained weights
• awesome-python-machine-learning-resources - GitHub - 9% open · ⏱️ 24.08.2022): (图像数据与CV)
• awesome-machine-learning-resources - **[List, Practice - image-models?style=social) (Table of Contents)
• awesome-list - pytorch-image-models - A collection of CV models, scripts, pretrained weights, based on PyTorch. (Computer Vision / General Purpose CV)
• awesome-cv - PyTorch image models