timm documentation

PP-HGNet

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

PP-HGNet

Kategori: CNN · Kod: timm/models/hgnet.py · Mimariler: 10 · Önceden eğitilmiş ağırlıklar: 19 (19 tanesi ImageNet-1k sonuçlarıyla)

PaddleClas'tan PP-HGNet ve PP-HGNetV2 (High Performance GPU Net), bir darmağana (convolution) dizisinin çıktıları birleştiren HG bloklarından oluşturulmuştur. ssld ağırlıkları, PaddleClas SSLD damıtma (distillation) yöntemiyle eğitilmiştir.

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

Model API desteği

ÖzellikHighPerfGpuNet
features_only=True özellik haritaları✓ (adım 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 (HighPerfGpuNet için hgnetv2_b0 224 boyutunda) çalıştırılarak kontrol edilir; 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
hgnet_base71.5825.142241
hgnet_small24.368.532242
hgnet_tiny14.744.542242
hgnetv2_b06.000.332242
hgnetv2_b16.340.492242
hgnetv2_b211.221.152242
hgnetv2_b316.291.782242
hgnetv2_b419.802.752242
hgnetv2_b539.576.562242
hgnetv2_b675.2616.882242

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ıklar kaynağa göre gruplandırılır: önceden eğitilmiş yapılandırmada belirtilen makale veya köken, aksi halde timm dışında bir Hub kuruluşu, aksi halde önceden eğitilmiş etiketin ilk kısmı (architecture.tag). Tek bir ağırlığı olan kaynaklar other altında listelenir.

ssld etiketleri

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
hgnet_base.ssld_in1k224 (test 288)85.4882881000apache-2.0timm
hgnet_small.ssld_in1k224 (test 288)84.3682881000apache-2.0timm
hgnet_tiny.ssld_in1k224 (test 288)82.5282881000apache-2.0timm
hgnetv2_b0.ssld_stage1_in22k_in1k224 (test 288)78.0382881000apache-2.0timm
hgnetv2_b0.ssld_stage2_ft_in1k224 (test 288)78.5902881000apache-2.0timm
hgnetv2_b1.ssld_stage1_in22k_in1k224 (test 288)79.0502881000apache-2.0timm
hgnetv2_b1.ssld_stage2_ft_in1k224 (test 288)79.9162881000apache-2.0timm
hgnetv2_b2.ssld_stage1_in22k_in1k224 (test 288)81.4562881000apache-2.0timm
hgnetv2_b2.ssld_stage2_ft_in1k224 (test 288)82.3622881000apache-2.0timm
hgnetv2_b3.ssld_stage1_in22k_in1k224 (test 288)83.1182881000apache-2.0timm
hgnetv2_b3.ssld_stage2_ft_in1k224 (test 288)83.5882881000apache-2.0timm
hgnetv2_b4.ssld_stage1_in22k_in1k224 (test 288)83.4502881000apache-2.0timm
hgnetv2_b4.ssld_stage2_ft_in1k224 (test 288)83.9522881000apache-2.0timm
hgnetv2_b5.ssld_stage1_in22k_in1k224 (test 288)84.9302881000apache-2.0timm
hgnetv2_b5.ssld_stage2_ft_in1k224 (test 288)85.1582881000apache-2.0timm
hgnetv2_b6.ssld_stage1_in22k_in1k224 (test 288)86.3002881000apache-2.0timm
hgnetv2_b6.ssld_stage2_ft_in1k224 (test 288)86.3802881000apache-2.0timm

paddle etiketleri

AğırlıkGirdi boyutuEn iyi 1 (%)Değerlendirme boyutuSınıflarLisansHub
hgnet_small.paddle_in1k224 (test 288)82.2222881000apache-2.0timm
hgnet_tiny.paddle_in1k224 (test 288)80.6442881000apache-2.0timm
Update on GitHub