timm documentation

VGG

Hugging Face's logo
Join the Hugging Face community

and get access to the augmented documentation experience

to get started

VGG

类别: CNN · 代码: timm/models/vgg.py · 架构: 8 · 预训练权重: 8(8 个具有 ImageNet-1k 结果)

由堆叠的 3x3 卷积构成的 VGG 网络,改编自 torchvision,并添加了 timm 功能。

论文

用法

请参阅快速入门了解如何加载预训练权重、进行图像预处理和微调,以及查阅特征提取了解如何将模型用作骨干网络。

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

模型 API 支持

功能VGG
features_only=True 特征图✓(步长 1、2、4、8、16、32)
forward_intermediates()✓
prune_intermediate_layers()✓
任意输入尺寸,无需更改✓
set_input_size()–
dynamic_img_size=True–
set_grad_checkpointing()✗
逐层学习率衰减(group_matcher())✓
reset_classifier()✓

✓ 支持,✓ (square) 仅支持方形尺寸,✗ 不支持,– 无需支持(模型本身已接受任意输入尺寸)。通过在每个功能上运行随机初始化的模型(VGG 使用 vgg11,尺寸为 224)进行验证,预期同一类的其他架构表现相同。详情请参阅 Model API 页面。

架构

架构参数(M)GMAC输入大小预训练权重
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

GMACs 和参数量对应于每种架构的默认输入大小和分类器,数据来自基准测试结果。

预训练权重

Top-1 是 results-imagenet.csv 中在所列评估规模下的 ImageNet-1k 验证准确率,- 表示该权重没有 ImageNet-1k 结果(例如预训练权重或图文权重)。Classes 是预训练分类器的大小,对于没有分类器的权重则为 none。

权重输入大小Top-1(%)评估大小类许可证Hub
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