get_vgg¶
-
get_vgg(num_layers, pretrained=False, ctx=cpu(0), root='/var/lib/jenkins/.mxnet/models', **kwargs)¶ 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.