timm documentation

NaFlex ViT

You are viewing main version, which requires installation from source. If you'd like regular pip install, checkout the latest stable version (v1.0.30).
Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

NaFlex ViT

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

Patch gömme ve konum kodlamayı tek bir modülde barındıran, değişken en-boy oranı ve çözünürlüğe sahip önceden patch'lenmiş girdileri (NaFlex), FlexiViT değişken patch boyutunu ve NaViT faktörize konum gömmelerini destekleyen bir Vision Transformer.

Makaleler

Önceden eğitilmiş ağırlıklarla referans alınır:

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

Model API desteği

ÖzellikNaFlexVit
features_only=True özellik haritaları✓ (adım 16, 16, 16)
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 (NaFlexVit, 384 boyutunda naflexvit_base_patch16_parfac_gap), 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
naflexvit_base_patch16_gap86.6355.863841
naflexvit_base_patch16_map93.7256.233840
naflexvit_base_patch16_par_gap86.6355.863841
naflexvit_base_patch16_parfac_gap86.4655.863841
naflexvit_base_patch16_radio_v398.25-5121
naflexvit_base_patch16_siglip92.9356.123841
naflexvit_giant_patch16_radio_v31159.62-5121
naflexvit_huge_patch16_radio_v3651.64-5121
naflexvit_huge_patch16_radio_v4651.65-5121
naflexvit_large_patch16_radio_v3319.93-5121
naflexvit_so150m2_patch16_reg1_gap136.0689.533840
naflexvit_so150m2_patch16_reg1_map142.4690.333840
naflexvit_so400m_patch16_radio_v4431.24-5121
naflexvit_so400m_patch16_siglip427.89259.653841

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.

arXiv:2412.07679

Önceden eğitilmiş yapılandırmalar arXiv:2412.07679 makalesine atıfta bulunur.

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
naflexvit_base_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_giant_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_huge_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_large_patch16_radio_v3.c_radio512--nonenvidia-open-model-licensetimm

v2 etiketleri

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
naflexvit_base_patch16_siglip.v2_webli384--noneapache-2.0timm
naflexvit_so400m_patch16_siglip.v2_webli384--noneapache-2.0timm

arXiv:2601.17237

Önceden eğitilmiş yapılandırmalar arXiv:2601.17237 makalesine atıfta bulunur.

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
naflexvit_huge_patch16_radio_v4.c_radio512--nonenvidia-open-model-licensetimm
naflexvit_so400m_patch16_radio_v4.c_radio512--nonenvidia-open-model-licensetimm

Diğer

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
naflexvit_base_patch16_gap.e300_s576_in1k38483.3403841000apache-2.0timm
naflexvit_base_patch16_par_gap.e300_s576_in1k38483.5083841000apache-2.0timm
naflexvit_base_patch16_parfac_gap.e300_s576_in1k38483.4683841000apache-2.0timm
Update on GitHub