备注

您正在阅读 MMClassification 0.x 版本的文档。MMClassification 0.x 会在 2022 年末被切换为次要分支。建议您升级到 MMClassification 1.0 版本，体验更多新特性和新功能。请查阅 MMClassification 1.0 的安装教程、迁移教程以及更新日志。

mmcls.apis.inference_model

mmcls.apis.inference_model(model, img)

Inference image(s) with the classifier.

参数

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

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

返回

The classification results that contains

class_name, pred_label and pred_score.

返回类型

result (dict)