mxnet.gluon.model_zoo.vision.get_vgg¶
-
mxnet.gluon.model_zoo.vision.get_vgg(num_layers, pretrained=False, ctx=cpu(0), root='/var/lib/jenkins/.mxnet/models', **kwargs)[source]¶ VGG model from the “Very Deep Convolutional Networks for Large-Scale Image Recognition” paper.
- Parameters
num_layers (int) – Number of layers for the variant of densenet. Options are 11, 13, 16, 19.
pretrained (bool, default False) – Whether to load the pretrained weights for model.
ctx (Context, default CPU) – The context in which to load the pretrained weights.
root (str, default $MXNET_HOME/models) – Location for keeping the model parameters.