timm documentation

GhostNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

GhostNet

Kategori: Efficient/Mobile · Kode: timm/models/ghostnet.py · Arsitektur: 10 · Bobot terlatih: 5 (5 dengan hasil ImageNet-1k)

GhostNet menghasilkan sebagian peta fiturnya dengan operasi murah (modul Ghost). GhostNetV2 menambahkan perhatian DFC untuk dependensi jarak jauh, sedangkan GhostNetV3 menggunakan re-parameterisasi dan distilasi pengetahuan dalam pelatihan.

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

Dukungan API Model

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

Arsitektur

ArsitekturParameter (M)GMACsUkuran inputBobot terlatih
ghostnet_0502.590.052240
ghostnet_1005.180.152241
ghostnet_1307.360.242240
ghostnetv2_1006.160.182241
ghostnetv2_1308.960.282241
ghostnetv2_16012.390.422241
ghostnetv3_0502.850.052240
ghostnetv3_1006.150.172241
ghostnetv3_1308.950.282240
ghostnetv3_16012.380.412240

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
ghostnet_100.in1k22473.9582241000apache-2.0timm
ghostnetv2_100.in1k22475.1862241000apache-2.0timm
ghostnetv2_130.in1k22476.7462241000apache-2.0timm
ghostnetv2_160.in1k22477.8362241000apache-2.0timm
ghostnetv3_100.in1k22476.9362241000apache-2.0timm
Update on GitHub