timm documentation

FocalNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

FocalNet

Kategori: Hibrida · Kode: timm/models/focalnet.py · Arsitektur: 12 · Bobot pralatih: 12 (6 dengan hasil ImageNet-1k)

Focal Modulation Networks (FocalNet) mengganti self-attention dengan focal modulation, yang mengagregasi konteks pada beberapa granularitas menggunakan depthwise convolutions dan mengaturnya ke setiap query token.

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

Dukungan API Model

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

Arsitektur

ArsitekturParameter (M)GMACsUkuran inputBobot terlatih
focalnet_base_lrf88.7515.432241
focalnet_base_srf88.1515.282241
focalnet_huge_fl3745.28118.262241
focalnet_huge_fl4686.46118.902241
focalnet_large_fl3239.13105.063841
focalnet_large_fl4239.32105.203841
focalnet_small_lrf50.348.742241
focalnet_small_srf49.898.622241
focalnet_tiny_lrf28.654.492241
focalnet_tiny_srf28.434.422241
focalnet_xlarge_fl3408.79185.613841
focalnet_xlarge_fl4409.03185.793841

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.

BobotUkuran inputTop-1 (%)Ukuran evaluasiKelasLisensiHub
focalnet_base_lrf.ms_in1k22483.8282241000mittimm
focalnet_base_srf.ms_in1k22483.8082241000mittimm
focalnet_huge_fl3.ms_in22k224--21842mittimm
focalnet_huge_fl4.ms_in22k224--nonemittimm
focalnet_large_fl3.ms_in22k384--21842mittimm
focalnet_large_fl4.ms_in22k384--21842mittimm
focalnet_small_lrf.ms_in1k22483.5082241000mittimm
focalnet_small_srf.ms_in1k22483.4282241000mittimm
focalnet_tiny_lrf.ms_in1k22482.1742241000mittimm
focalnet_tiny_srf.ms_in1k22482.1462241000mittimm
focalnet_xlarge_fl3.ms_in22k384--21842mittimm
focalnet_xlarge_fl4.ms_in22k384--21842mittimm
Update on GitHub