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