PyTorch Image Models

https://github.com/rwightman/pytorch-image-models

AI & ML interests

Computer Vision

Recent Activity

rwightman new activity 12 days ago

timm/vit_small_patch16_224.augreg_in1k:Exact timm Train arguments

rwightman new activity 25 days ago

timm/inception_next_atto.sail_in1k:Improve model card

pcuenq authored a paper 27 days ago

SmolVLM: Redefining small and efficient multimodal models

View all activity

Organization Card

Community About org cards

timm

PyTorch Image Models (timm) is a collection of image models, layers, utilities, optimizers, schedulers, data-loaders / augmentations, and reference training / validation scripts that aim to pull together a wide variety of SOTA models with ability to reproduce ImageNet training results.

Find all the timm models here.
timm github repository
timm documentation 📚

Collections 16

spaces 5

Timm + Transformers

Any timm model with the transformers integration

JupyterLab (timm)

Improved JuptyerLab template w/ timm

The timm Leaderboard

Display and analyze PyTorch Image Models leaderboard

timm Attention Visualization

Visualize attention maps for images using selected models

timm CAM

models 1,595

timm/coat_lite_medium_384.in1k

Image Classification • Updated Mar 18 • 240

timm/coat_small.in1k

Image Classification • Updated Mar 18 • 724

timm/ViT-gopt-16-SigLIP2-384

Zero-Shot Image Classification • Updated Feb 21 • 1.53k • 1

timm/ViT-gopt-16-SigLIP2-256

Zero-Shot Image Classification • Updated Feb 21 • 41.1k

timm/ViT-SO400M-16-SigLIP2-512

Zero-Shot Image Classification • Updated Feb 21 • 1.03k • 4

timm/ViT-SO400M-16-SigLIP2-384

Zero-Shot Image Classification • Updated Feb 21 • 95.5k • 2

timm/ViT-SO400M-16-SigLIP2-256

Zero-Shot Image Classification • Updated Feb 21 • 940

timm/ViT-SO400M-14-SigLIP2-378

Zero-Shot Image Classification • Updated Feb 21 • 1.82k • 1

timm/ViT-SO400M-14-SigLIP2

Zero-Shot Image Classification • Updated Feb 21 • 1.53k

timm/ViT-L-16-SigLIP2-512

Zero-Shot Image Classification • Updated Feb 21 • 66 • 2

datasets 15

timm/plant-pathology-2021

Viewer • Updated Nov 22, 2024 • 18.6k • 638 • 4

timm/mini-imagenet

Viewer • Updated Nov 20, 2024 • 65k • 3.83k • 9

timm/imagenet-22k-wds

Viewer • Updated Jan 29, 2024 • 123k • 664 • 13

timm/imagenet-12k-wds

Viewer • Updated Jan 7, 2024 • 127k • 537 • 8

timm/imagenet-w21-webp-wds

Viewer • Updated Jan 7, 2024 • 113k • 26 • 3

timm/imagenet-w21-wds

Viewer • Updated Jan 7, 2024 • 60.1k • 55 • 4

timm/imagenet-1k-wds

Viewer • Updated Jan 7, 2024 • 80.7k • 3.45k • 22

timm/imagenet-w21-p

Viewer • Updated Jan 7, 2024 • 11.6M • 120 • 4

timm/eurosat-rgb

Viewer • Updated Jan 7, 2024 • 27k • 98 • 1

timm/resisc45

Viewer • Updated Jan 7, 2024 • 31.5k • 455 • 1