timm documentation
VGG
快速入门
教程
模型页面
模型系列结果
卷积网络
对齐的 XceptionBYOBNet(GENet、RepVGG、MobileOne 等)ConvMixerConvNeXtCSPNet(CSPResNet、CSPDarkNet、DarkNet)CSPNeXtDenseNetDLADPNHRNetInception-ResNet-v2Inception-v3Inception-v4InceptionNeXtMambaOutNASNetNFNetPNASNetPP-HGNetRDNetRegNetRes2NetResNeStResNetResNetV2 (BiT)SelecSLSSENet(旧版)SKNetTResNetVGGVoVNetXception(旧版)
视觉 Transformer
混合卷积 / 注意力网络
高效与移动网络
MLP 及其他 token 混合器
多模态模型的视觉编码器
参考
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 支持
✓ 支持,✓ (square) 仅支持方形尺寸,✗ 不支持,– 无需支持(模型本身已接受任意输入尺寸)。通过在每个功能上运行随机初始化的模型(VGG 使用 vgg11,尺寸为 224)进行验证,预期同一类的其他架构表现相同。详情请参阅 Model API 页面。
架构
| 架构 | 参数(M) | GMAC | 输入大小 | 预训练权重 |
|---|---|---|---|---|
vgg11 | 132.86 | 7.61 | 224 | 1 |
vgg11_bn | 132.87 | 7.62 | 224 | 1 |
vgg13 | 133.05 | 11.31 | 224 | 1 |
vgg13_bn | 133.05 | 11.33 | 224 | 1 |
vgg16 | 138.36 | 15.47 | 224 | 1 |
vgg16_bn | 138.37 | 15.50 | 224 | 1 |
vgg19 | 143.67 | 19.63 | 224 | 1 |
vgg19_bn | 143.68 | 19.66 | 224 | 1 |
GMACs 和参数量对应于每种架构的默认输入大小和分类器,数据来自基准测试结果。
预训练权重
Top-1 是 results-imagenet.csv 中在所列评估规模下的 ImageNet-1k 验证准确率,- 表示该权重没有 ImageNet-1k 结果(例如预训练权重或图文权重)。Classes 是预训练分类器的大小,对于没有分类器的权重则为 none。
| 权重 | 输入大小 | Top-1(%) | 评估大小 | 类 | 许可证 | Hub |
|---|---|---|---|---|---|---|
vgg11.tv_in1k | 224 | 69.050 | 224 | 1000 | bsd-3-clause | timm |
vgg11_bn.tv_in1k | 224 | 70.374 | 224 | 1000 | bsd-3-clause | timm |
vgg13.tv_in1k | 224 | 69.950 | 224 | 1000 | bsd-3-clause | timm |
vgg13_bn.tv_in1k | 224 | 71.560 | 224 | 1000 | bsd-3-clause | timm |
vgg16.tv_in1k | 224 | 71.594 | 224 | 1000 | bsd-3-clause | timm |
vgg16_bn.tv_in1k | 224 | 73.354 | 224 | 1000 | bsd-3-clause | timm |
vgg19.tv_in1k | 224 | 72.400 | 224 | 1000 | bsd-3-clause | timm |
vgg19_bn.tv_in1k | 224 | 74.230 | 224 | 1000 | bsd-3-clause | timm |