timm documentation

HRNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

HRNet

Kategori: CNN · Kode: timm/models/hrnet.py · Arsitektur: 11 · Bobot terlatih: 14 (14 dengan hasil ImageNet-1k)

Jaringan Resolusi Tinggi mempertahankan cabang resolusi tinggi di seluruh jaringan dan menambahkan cabang resolusi lebih rendah secara paralel, dengan bertukar informasi di antara keduanya secara berulang.

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

Dukungan API Model

FiturHighResolutionNet
feature maps features_only=True✓ (strides 2, 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 (HighResolutionNet dengan hrnet_w18_small pada 224), arsitektur lain dari kelas yang sama diharapkan berperilaku sama. Lihat halaman Model API untuk detailnya.

Arsitektur

ArsitekturParameter (M)GMACsUkuran inputBobot terlatih
hrnet_w1821.304.322242
hrnet_w18_small13.191.612242
hrnet_w18_small_v215.602.622242
hrnet_w18_ssld21.304.322241
hrnet_w3037.718.152241
hrnet_w3241.238.972241
hrnet_w4057.5612.752241
hrnet_w4467.0614.942241
hrnet_w4877.4717.342241
hrnet_w48_ssld77.4717.342241
hrnet_w64128.0628.972241

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.

ms tags

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
hrnet_w18.ms_aug_in1k22478.1102241000mittimm
hrnet_w18.ms_in1k22476.7702241000mittimm
hrnet_w18_small.ms_in1k22472.3402241000mittimm
hrnet_w18_small_v2.ms_in1k22475.0942241000mittimm
hrnet_w30.ms_in1k22478.2022241000mittimm
hrnet_w32.ms_in1k22478.4522241000mittimm
hrnet_w40.ms_in1k22478.9282241000mittimm
hrnet_w44.ms_in1k22478.8902241000mittimm
hrnet_w48.ms_in1k22479.3282241000mittimm
hrnet_w64.ms_in1k22479.4582241000mittimm

gluon tags

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
hrnet_w18_small.gluon_in1k22473.9342241000apache-2.0timm
hrnet_w18_small_v2.gluon_in1k22478.2022241000apache-2.0timm

tag paddle

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
hrnet_w18_ssld.paddle_in1k224 (uji 288)82.0522881000mittimm
hrnet_w48_ssld.paddle_in1k224 (uji 288)84.4842881000mittimm
Update on GitHub