Table Of Contents
Table Of Contents

mxnet.gluon.model_zoo.vision.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