timm documentation

GCViT

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

GCViT

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

Global Context Vision Transformers (GCViT), yerel pencere dikkati ile global bağlam dikkatini değiştirerek kullanır; global bağlam dikkatinin sorguları, konvolüsyonel bir modül tarafından tüm özellik haritasından üretilir.

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

Model API desteği

ÖzellikGlobalContextVit
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 (GlobalContextVit için gcvit_xxtiny, 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
gcvit_base90.3214.872241
gcvit_small51.098.572241
gcvit_tiny28.224.792241
gcvit_xtiny19.982.932241
gcvit_xxtiny12.002.142241

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
gcvit_base.in1k22484.4562241000apache-2.0orijinal
gcvit_small.in1k22483.9042241000apache-2.0orijinal
gcvit_tiny.in1k22483.4002241000apache-2.0orijinal
gcvit_xtiny.in1k22481.9642241000apache-2.0orijinal
gcvit_xxtiny.in1k22479.7562241000apache-2.0orijinal

Atıf

@article{hatamizadeh2022global,
  title={Global Context Vision Transformers},
  author={Hatamizadeh, Ali and Yin, Hongxu and Kautz, Jan and Molchanov, Pavlo},
  journal={arXiv preprint arXiv:2206.09959},
  year={2022}
}
Update on GitHub