PyTorch Image Models

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

AI & ML interests

Computer Vision

Recent Activity

rwightman new activity 1 day ago

timm/vit_pwee_patch16_reg1_gap_256.sbb_in1k:Update train_hparams.yaml

rwightman updated a model 7 days ago

timm/vit_pe_core_gigantic_patch14_448.fb

rwightman updated a model 7 days ago

timm/vit_pe_core_base_patch16_224.fb

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

View 16 collections

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,634

timm/vit_pwee_patch16_reg1_gap_256.sbb_in1k

Image Classification • Updated 1 day ago • 1.12k • 2

timm/vit_pe_core_gigantic_patch14_448.fb

Image Feature Extraction • Updated 7 days ago • 28

timm/vit_pe_core_base_patch16_224.fb

Image Feature Extraction • Updated 7 days ago • 507 • 1

timm/vit_pe_core_large_patch14_336.fb

Image Feature Extraction • Updated 7 days ago • 186

timm/vit_pe_spatial_gigantic_patch14_448.fb

Image Feature Extraction • Updated 7 days ago • 21

timm/vit_pe_lang_large_patch14_448.fb

Image Feature Extraction • Updated 7 days ago • 191 • 1

timm/vit_pe_lang_gigantic_patch14_448.fb

Image Feature Extraction • Updated 7 days ago • 23

timm/naflexvit_base_patch16_parfac_gap.e300_s576_in1k

Image Classification • Updated 8 days ago • 41 • 2

timm/naflexvit_base_patch16_par_gap.e300_s576_in1k

Image Classification • Updated 8 days ago • 49 • 2

timm/naflexvit_base_patch16_gap.e300_s576_in1k

Image Classification • Updated 8 days ago • 42 • 1

View 1,634 models

datasets 15

timm/plant-pathology-2021

Viewer • Updated Nov 22, 2024 • 18.6k • 90 • 5

timm/mini-imagenet

Viewer • Updated Nov 20, 2024 • 65k • 2.35k • 11

timm/imagenet-22k-wds

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

timm/imagenet-12k-wds

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

timm/imagenet-w21-webp-wds

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

timm/imagenet-w21-wds

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

timm/imagenet-1k-wds

Viewer • Updated Jan 7, 2024 • 80.7k • 5.75k • 25

timm/imagenet-w21-p

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

timm/eurosat-rgb

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

timm/resisc45

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

View 15 datasets