timm documentation

DaViT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

DaViT

Kategori: ViT · Kod: timm/models/davit.py · Mimariler: 8 · Önceden eğitilmiş ağırlıklar: 5 (3'ü ImageNet-1k sonuçlarıyla)

Dual Attention Vision Transformers (DaViT), hiyerarşik bir tasarımda, token sayısına lineer olan uzamsal pencere dikkati ve kanal grubu dikkatini birleştirir.

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('davit_tiny.msft_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Model API desteği

ÖzellikDaVit
features_only=True özellik haritaları✓ (adım 4, 8, 16, 32)
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 (DaVit için davit_tiny 224 boyutunda); 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
davit_base87.9515.512241
davit_base_fl90.37190.327681
davit_giant1406.47192.922240
davit_huge348.9261.232240
davit_huge_fl360.64744.847681
davit_large196.8134.602240
davit_small49.758.802241
davit_tiny28.364.542241

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ıklar kaynağa göre gruplandırılır: önceden eğitilmiş yapılandırmada belirtilen makale veya köken, aksi halde timm dışında bir Hub kuruluşu, aksi halde önceden eğitilmiş etiketin ilk kısmı (architecture.tag). Tek bir ağırlığı olan kaynaklar other altında listelenir.

msft etiketleri

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
davit_base.msft_in1k22484.6522241000apache-2.0timm
davit_small.msft_in1k22484.2522241000apache-2.0timm
davit_tiny.msft_in1k22482.6942241000apache-2.0timm

microsoft

Hugging Face Hub'da microsoft tarafından barındırılır.

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
davit_base_fl.msft_florence2768--noneapache-2.0microsoft/Florence-2-base
davit_huge_fl.msft_florence2768--noneapache-2.0microsoft/Florence-2-large
Update on GitHub