🖼️

timm (PyTorch Image Models)

Computer Vision

600+ pretrained vision models for PyTorch.

31kstars4.6kforksPython

About

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

Key Features

Pretrained models
Fine-tuning
Training scripts
Benchmarks

Related Resources

👁️

YOLO (Ultralytics)

Real-time object detection system.

🖼️

OpenCLIP

Open-source implementation of CLIP models.

✂️

Segment Anything Model

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