vgg16

paddle.vision.models. vgg16 ( pretrained=False, batch_norm=False, **kwargs ) [source]

VGG 16-layer model

Parameters
  • pretrained (bool) – If True, returns a model pre-trained on ImageNet. Default: False.

  • batch_norm (bool) – If True, returns a model with batch_norm layer. Default: False.

Examples

from paddle.vision.models import vgg16

# build model
model = vgg16()

# build vgg16 model with batch_norm
model = vgg16(batch_norm=True)