timm documentation

Hiera

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

Hiera

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

Hiera adalah vision transformer hierarkis tanpa modul khusus (konvolusi, jendela bergeser, bias posisi relatif), yang mengandalkan pra-pelatihan masked autoencoder (MAE) untuk mempelajari bias spasial.

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

Dukungan API Model

FiturHiera
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 (Hiera dengan hiera_tiny_224 pada 224), arsitektur lain dari kelas yang sama diharapkan berperilaku sama. Lihat halaman Model API untuk detailnya.

Arsitektur

ArsitekturParameter (M)GMACsUkuran inputBobot terlatih
hiera_base_22451.529.402242
hiera_base_abswin_25651.2712.462560
hiera_base_plus_22469.9012.672242
hiera_huge_224672.78124.852242
hiera_large_224213.7440.342242
hiera_small_22435.016.422242
hiera_small_abswin_25634.368.292564
hiera_tiny_22427.914.912242

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.

mae tag

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
hiera_base_224.mae224--nonecc-by-nc-4.0timm
hiera_base_224.mae_in1k_ft_in1k22484.5282241000cc-by-nc-4.0timm
hiera_base_plus_224.mae224--nonecc-by-nc-4.0timm
hiera_base_plus_224.mae_in1k_ft_in1k22485.1362241000cc-by-nc-4.0timm
hiera_huge_224.mae224--nonecc-by-nc-4.0timm
hiera_huge_224.mae_in1k_ft_in1k22486.8402241000cc-by-nc-4.0timm
hiera_large_224.mae224--nonecc-by-nc-4.0timm
hiera_large_224.mae_in1k_ft_in1k22486.0462241000cc-by-nc-4.0timm
hiera_small_224.mae224--nonecc-by-nc-4.0timm
hiera_small_224.mae_in1k_ft_in1k22483.8882241000cc-by-nc-4.0timm
hiera_tiny_224.mae224--nonecc-by-nc-4.0timm
hiera_tiny_224.mae_in1k_ft_in1k22482.8262241000cc-by-nc-4.0timm

tag sbb2

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
hiera_small_abswin_256.sbb2_e200_in12k256--11821apache-2.0timm
hiera_small_abswin_256.sbb2_e200_in12k_ft_in1k25684.9122561000apache-2.0timm
hiera_small_abswin_256.sbb2_pd_e200_in12k256--11821apache-2.0timm
hiera_small_abswin_256.sbb2_pd_e200_in12k_ft_in1k25684.5642561000apache-2.0timm
Update on GitHub