Table Of Contents
Table Of Contents

get_model

mxnet.gluon.model_zoo.vision.get_model(name, **kwargs)[source]

Returns a pre-defined model by name

Parameters:
  • name (str) – Name of the model.
  • pretrained (bool) – Whether to load the pretrained weights for model.
  • classes (int) – Number of classes for the output layer.
  • 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.
Returns:

The model.

Return type:

HybridBlock