timm documentation
VGG
VGG
Category: CNN · Code: timm/models/vgg.py · Architectures: 8 · Pretrained weights: 8 (8 with ImageNet-1k results)
VGG networks of stacked 3x3 convolutions, adapted from torchvision with timm functionality added.
Papers
Usage
See the Quickstart for loading pretrained weights, image preprocessing and fine-tuning, and Feature Extraction for using the models as backbones.
>>> import timm
>>> model = timm.create_model('vgg11.tv_in1k', pretrained=True)
>>> data_config = timm.data.resolve_model_data_config(model)Model API support
✓ supported, ✓ (square) works for square sizes only, ✗ not supported, – not needed (the model already accepts any input
size). Checked by running each feature on a randomly initialized model (VGG with vgg11 at 224), other architectures
of the same class are expected to behave the same. See the Model API page for details.
Architectures
| Architecture | Params (M) | GMACs | Input size | Pretrained weights |
|---|---|---|---|---|
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 and parameter counts are for the default input size and classifier of each architecture, from the benchmark results.
Pretrained weights
Top-1 is ImageNet-1k validation accuracy from results-imagenet.csv at
the listed eval size, - if the weight has no ImageNet-1k result (e.g. pretraining or image-text weights). Classes is the
size of the pretrained classifier, none for weights without one.
| Weight | Input size | Top-1 (%) | Eval size | Classes | License | 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 |