VGG
27 memberslucid.models.vision.vggVGG family — Simonyan & Zisserman, 2014.
Simonyan, Karen, and Andrew Zisserman. "Very Deep Convolutional Networks for Large-Scale Image Recognition." International Conference on Learning Representations (ICLR), 2015.
VGG demonstrated that uniformity and depth — rather than bespoke layer designs — were the dominant factors driving image- classification accuracy in 2014. The network is built from a single recipe: stack convolutions with stride 1 and padding 1, periodically halve the spatial resolution with a max-pool, double the channel count at each pooling boundary, and end with three large fully-connected layers.
The kernel choice is theoretically motivated. Two stacked convolutions have the same receptive field as a single convolution but use only parameters versus , while inserting an extra nonlinearity between them. Three stacked layers match a receptive field at versus parameters and two extra nonlinearities. Depth at a fixed receptive field therefore buys representational power essentially for free.
The paper's variants (configurations A, B, D, E corresponding to VGG-11/13/16/19) differ only in how many convs are stacked between pooling layers. Despite a parameter count of 138 M — most of it locked up in the 4096-dim fully-connected layers — VGG-16 achieves a top-5 ImageNet error of 7.3%, second only to GoogLeNet in ILSVRC-2014, and its features transferred so well that "VGG features" became the standard backbone for downstream tasks (object detection, segmentation, style transfer) for years after publication.
Classes
Functions
vgg_11→ VGGVGG-11 feature-extracting backbone (config A, no BatchNorm).
vgg_11_bn→ VGGVGG-11 with BatchNorm — feature-extracting backbone.
vgg_11_bn_cls→ VGGForImageClassificationVGG-11 with BatchNorm — image classifier.
vgg_11_cls→ VGGForImageClassificationVGG-11 image classifier (config A, no BatchNorm).
vgg_13→ VGGVGG-13 feature-extracting backbone (config B, no BatchNorm).
vgg_13_bn→ VGGVGG-13 with BatchNorm — feature-extracting backbone.
vgg_13_bn_cls→ VGGForImageClassificationVGG-13 with BatchNorm — image classifier.
vgg_13_cls→ VGGForImageClassificationVGG-13 image classifier (config B, no BatchNorm).
vgg_16→ VGGVGG-16 feature-extracting backbone (config D, no BatchNorm).
vgg_16_bn→ VGGVGG-16 with BatchNorm — feature-extracting backbone.
vgg_16_bn_cls→ VGGForImageClassificationVGG-16 with BatchNorm — image classifier.
vgg_16_cls→ VGGForImageClassificationVGG-16 image classifier (config D, no BatchNorm).
vgg_19→ VGGVGG-19 feature-extracting backbone (config E, no BatchNorm).
vgg_19_bn→ VGGVGG-19 with BatchNorm — feature-extracting backbone.
vgg_19_bn_cls→ VGGForImageClassificationVGG-19 with BatchNorm — image classifier.
vgg_19_cls→ VGGForImageClassificationVGG-19 image classifier (config E, no BatchNorm).
Weights
VGG11BNWeightsPretrained weights for lucid.models.vgg_11_bn_cls.
VGG11WeightsPretrained weights for lucid.models.vgg_11_cls.
VGG13BNWeightsPretrained weights for lucid.models.vgg_13_bn_cls.
VGG13WeightsPretrained weights for lucid.models.vgg_13_cls.
VGG16BNWeightsPretrained weights for lucid.models.vgg_16_bn_cls.
VGG16WeightsPretrained weights for lucid.models.vgg_16_cls.
VGG19BNWeightsPretrained weights for lucid.models.vgg_19_bn_cls.
VGG19WeightsPretrained weights for lucid.models.vgg_19_cls.