Shortcuts

Note

You are reading the documentation for MMClassification 0.x, which will soon be deprecated at the end of 2022. We recommend you upgrade to MMClassification 1.0 to enjoy fruitful new features and better performance brought by OpenMMLab 2.0. Check the installation tutorial, migration tutorial and changelog for more details.

mmcls.apis.inference_model

mmcls.apis.inference_model(model, img)[source]

Inference image(s) with the classifier.

Parameters
  • model (nn.Module) – The loaded classifier.

  • img (str/ndarray) – The image filename or loaded image.

Returns

The classification results that contains

class_name, pred_label and pred_score.

Return type

result (dict)

Read the Docs v: master
Versions
master
latest
1.x
dev-1.x
Downloads
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.