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 · Kod: timm/models/ghostnet.py · Mimariler: 10 · Önceden eğitilmiş ağırlıklar: 5 (5 tanesi ImageNet-1k sonuçlarıyla)

GhostNet, özellik haritalarının bir kısmını ucuz işlemlerle (Ghost modülleri) üretir. GhostNetV2, uzun menzilli bağımlılıklar için DFC dikkatini ekler; GhostNetV3 ise eğitim sırasında yeniden parametrelendirme ve bilgi damıtma kullanır.

Makaleler

Kullanım

Önceden eğitilmiş ağırlıkları yükleme, görüntü ön işleme ve ince ayar için Hızlı Başlangıç bölümüne, modelleri omurga olarak kullanmak için ise Özellik Çıkarımı bölümüne bakın.

>>> import timm
>>> model = timm.create_model('ghostnetv2_160.in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)

Model API desteği

ÖzellikGhostNet
features_only=True özellik haritaları✓ (adım 2, 4, 8, 16, 32)
forward_intermediates()✓
prune_intermediate_layers()✓
Herhangi bir giriş boyutu, değişiklik gerekmez✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✓
Katman bazlı LR azaltma (group_matcher())✓
reset_classifier()✓

✓ destekleniyor, ✓ (kare) yalnızca kare boyutlarda çalışır, ✗ desteklenmiyor, – gerekli değil (model zaten herhangi bir giriş boyutunu kabul eder). Her özellik, rastgele başlatılmış bir model üzerinde çalıştırılarak kontrol edilir (GhostNet için ghostnet_050, 224 boyutunda); aynı sınıftaki diğer mimarilerin de aynı şekilde davranması beklenir. Ayrıntılar için Model API sayfasına bakın.

Mimariler

MimariParametreler (M)GMAC'larGirdi boyutuÖnceden eğitilmiş ağırlıklar
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

GMAC'lar ve parametre sayıları, her mimarinin varsayılan giriş boyutu ve sınıflandırıcısı içindir ve benchmark sonuçlarından alınmıştır.

Önceden eğitilmiş ağırlıklar

Top-1, belirtilen değerlendirme boyutunda results-imagenet.csv dosyasındaki ImageNet-1k doğrulama doğruluğudur; ağırlığın ImageNet-1k sonucu yoksa (ör. ön eğitim veya görsel-metin ağırlıkları) bu değer "-" olarak gösterilir. Classes, ön eğitilmiş sınıflandırıcının boyutudur; sınıflandırıcısı olmayan ağırlıklar için bu değer "none"dur.

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
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