Back to Resources
🖼️

timm (PyTorch Image Models)

Computer Vision

600+ pretrained vision models for PyTorch.

33kstars4.8kforksPython
View on GitHub

About

timm provides an extensive collection of pretrained image classification models (ViT, ResNet, EfficientNet, and many more) with a consistent training and inference API. Maintained by Hugging Face.

Key Features

  • Pretrained models
  • Fine-tuning
  • Training scripts
  • Benchmarks

Tags

ViTResNetClassificationPyTorch

Related Resources

👁️

YOLO (Ultralytics)

Real-time object detection, segmentation, and pose estimation.

✂️

Segment Anything Model

Meta's foundation model for zero-shot image segmentation.

📷

OpenCV

The world's largest computer vision library.