timm documentation
EfficientViM
EfficientViM
Kategori: Efficient/Mobile Β· Kode: timm/models/efficientvim.py Β· Arsitektur: 8 Β· Bobot terlatih: 12 (0 dengan hasil ImageNet-1k)
Efficient Vision Mamba (EfficientViM) dibangun di sekitar lapisan Hidden State Mixer based State Space Duality (HSM-SSD), yang diimplementasikan di sini dalam PyTorch murni. Model _dist menambahkan kepala distilasi.
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('efficientvim_m1.e300_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)Dukungan API Model
| Fitur | EfficientViM | EfficientViMDistilled |
|---|---|---|
feature maps features_only=True | β (strides 16, 32, 64) | β (strides 16, 32, 64) |
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 (EfficientViM dengan efficientvim_m1 pada 224, EfficientViMDistilled dengan efficientvim_m1_dist pada 224), arsitektur lain dari kelas yang sama diharapkan berperilaku sama. Lihat halaman Model API untuk detailnya.
Arsitektur
| Arsitektur | Kelas | Parameter (M) | GMACs | Ukuran input | Bobot terlatih |
|---|---|---|---|---|---|
efficientvim_m1 | EfficientViM | 6.68 | - | 224 | 2 |
efficientvim_m1_dist | EfficientViMDistilled | 7.64 | - | 224 | 1 |
efficientvim_m2 | EfficientViM | 13.90 | - | 224 | 2 |
efficientvim_m2_dist | EfficientViMDistilled | 15.32 | - | 224 | 1 |
efficientvim_m3 | EfficientViM | 16.60 | - | 224 | 2 |
efficientvim_m3_dist | EfficientViMDistilled | 18.18 | - | 224 | 1 |
efficientvim_m4 | EfficientViM | 19.61 | - | 256 | 2 |
efficientvim_m4_dist | EfficientViMDistilled | 21.18 | - | 256 | 1 |
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 | Ukuran input | Top-1 (%) | Ukuran evaluasi | Kelas | Lisensi | Hub |
|---|---|---|---|---|---|---|
efficientvim_m1.e300_in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m1.e450_in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m1_dist.in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m2.e300_in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m2.e450_in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m2_dist.in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m3.e300_in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m3.e450_in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m3_dist.in1k | 224 | - | - | 1000 | mit | timm |
efficientvim_m4.e300_in1k | 256 | - | - | 1000 | mit | timm |
efficientvim_m4.e450_in1k | 256 | - | - | 1000 | mit | timm |
efficientvim_m4_dist.in1k | 256 | - | - | 1000 | mit | timm |