timm documentation

ConvMixer

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

ConvMixer

Kategori: CNN · Kod: timm/models/convmixer.py · Mimariler: 3 · Önceden eğitilmiş ağırlıklar: 3 (3 tanesi ImageNet-1k sonuçlarıyla)

ConvMixer, görüntüyü yamalar (patches) olarak gömme (embed) işlemine tabi tutar ve bu çözünürlüğü baştan sona korur; derinlik içi konvolüsyonları (uzamsal karıştırma) ve nokta içi konvolüsyonları (kanal karıştırma) sırayla uygular.

Makaleler

Kullanım

Önceden eğitilmiş ağırlıkları yükleme, görüntü ön işleme ve ince ayar için Hızlı Başlangıç bölümüne, modelleri omurga olarak kullanmak için ise Özellik Çıkarımı bölümüne bakın.

>>> import timm
>>> model = timm.create_model('convmixer_1024_20_ks9_p14.in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Model API desteği

ÖzellikConvMixer
features_only=True özellik haritaları✓ (adım 7, 7, 7)
forward_intermediates()✓
prune_intermediate_layers()✓
Herhangi bir giriş boyutu, değişiklik gerekmez✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Katman bazlı LR azaltma (group_matcher())✓
reset_classifier()✓

✓ destekleniyor, ✓ (kare) yalnızca kare boyutlarda çalışır, ✗ desteklenmiyor, – gerekli değil (model zaten herhangi bir giriş boyutunu kabul eder). Her özellik, rastgele başlatılmış bir model üzerinde çalıştırılarak kontrol edilir (ConvMixer, 224 boyutunda convmixer_768_32), aynı sınıftaki diğer mimarilerin de aynı şekilde davranması beklenir. Ayrıntılar için Model API sayfasına bakın.

Mimariler

MimariParametreler (M)GMAC'larGirdi boyutuÖnceden eğitilmiş ağırlıklar
convmixer_768_3221.1119.552241
convmixer_1024_20_ks9_p1424.385.552241
convmixer_1536_2051.6348.682241

GMAC'lar ve parametre sayıları, her mimarinin varsayılan giriş boyutu ve sınıflandırıcısı içindir ve benchmark sonuçlarından alınmıştır.

Önceden eğitilmiş ağırlıklar

Top-1, belirtilen değerlendirme boyutunda results-imagenet.csv dosyasındaki ImageNet-1k doğrulama doğruluğudur; ağırlığın ImageNet-1k sonucu yoksa (ör. ön eğitim veya görsel-metin ağırlıkları) bu değer "-" olarak gösterilir. Classes, ön eğitilmiş sınıflandırıcının boyutudur; sınıflandırıcısı olmayan ağırlıklar için bu değer "none"dur.

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
convmixer_768_32.in1k22480.1582241000mittimm
convmixer_1024_20_ks9_p14.in1k22476.9442241000mittimm
convmixer_1536_20.in1k22481.3822241000mittimm
Update on GitHub