timm documentation

DaViT

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

and get access to the augmented documentation experience

to get started

DaViT

Kategori: ViT · Kode: timm/models/davit.py · Arsitektur: 8 · Bobot terlatih: 5 (3 dengan hasil ImageNet-1k)

Dual Attention Vision Transformers (DaViT) menggabungkan spatial window attention dan channel group attention, keduanya linear terhadap jumlah token, dalam desain hierarkis.

Makalah

Penggunaan

Lihat Quickstart untuk memuat bobot terlatih, prapemrosesan gambar, dan penyesuaian halus, serta Ekstraksi Fitur untuk menggunakan model sebagai tulang punggung.

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

Dukungan API Model

FiturDaVit
feature maps features_only=True✓ (strides 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Ukuran input apa pun, tidak perlu perubahan✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Penurunan LR per lapisan (group_matcher())✓
reset_classifier()✓

✓ didukung, ✓ (persegi) hanya berfungsi untuk ukuran persegi, ✗ tidak didukung, – tidak diperlukan (model sudah menerima ukuran input apa pun). Diperiksa dengan menjalankan setiap fitur pada model yang diinisialisasi secara acak (DaVit dengan davit_tiny pada 224), arsitektur lain dari kelas yang sama diharapkan berperilaku sama. Lihat halaman Model API untuk detailnya.

Arsitektur

ArsitekturParameter (M)GMACsUkuran inputBobot terlatih
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

GMACs dan jumlah parameter adalah untuk ukuran input dan klasifer default dari setiap arsitektur, dari hasil benchmark.

Bobot terlatih

Top-1 adalah akurasi validasi ImageNet-1k dari results-imagenet.csv pada ukuran eval yang tercantum, - jika bobot tidak memiliki hasil ImageNet-1k (misalnya bobot pretraining atau image-text). Classes adalah ukuran dari classifier pretrained, none untuk bobot yang tidak memilikinya.

Bobot dikelompokkan berdasarkan sumber: paper atau asal yang diberikan dalam konfigurasi pretrained, jika tidak ada maka organisasi Hub selain timm, jika tidak ada maka bagian pertama dari tag pretrained (architecture.tag). Sumber dengan satu bobot terdaftar di bawah other.

msft tags

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
davit_base.msft_in1k22484.6522241000apache-2.0timm
davit_small.msft_in1k22484.2522241000apache-2.0timm
davit_tiny.msft_in1k22482.6942241000apache-2.0timm

microsoft

Dihosting oleh microsoft di Hugging Face Hub.

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
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