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)

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)