timm documentation

VGG

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

VGG

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

torchvision'dan uyarlanan, timm işlevselliği eklenmiş, üst üste yığılmış 3x3 konvolüsyonlardan oluşan VGG ağları.

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

Model API desteği

ÖzellikVGG
features_only=True özellik haritaları✓ (adım 1, 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 (VGG ile vgg11 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
vgg11132.867.612241
vgg11_bn132.877.622241
vgg13133.0511.312241
vgg13_bn133.0511.332241
vgg16138.3615.472241
vgg16_bn138.3715.502241
vgg19143.6719.632241
vgg19_bn143.6819.662241

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
vgg11.tv_in1k22469.0502241000bsd-3-clausetimm
vgg11_bn.tv_in1k22470.3742241000bsd-3-clausetimm
vgg13.tv_in1k22469.9502241000bsd-3-clausetimm
vgg13_bn.tv_in1k22471.5602241000bsd-3-clausetimm
vgg16.tv_in1k22471.5942241000bsd-3-clausetimm
vgg16_bn.tv_in1k22473.3542241000bsd-3-clausetimm
vgg19.tv_in1k22472.4002241000bsd-3-clausetimm
vgg19_bn.tv_in1k22474.2302241000bsd-3-clausetimm
Update on GitHub