安装与使用¶
安装¶
MMEval 依赖 Python 3.6+,可以通过 pip 来安装 MMEval。安装 MMEval 的过程中会安装一些 MMEval 运行时的依赖库:
pip install mmeval
如果要安装 MMEval 中所有评测指标都需要的依赖,可以通过以下命令安装:
pip install 'mmeval[all]'
使用¶
MMEval 中的评测指标提供两种使用方式,以 Accuracy 为例:
from mmeval import Accuracy
import numpy as np
accuracy = Accuracy()
第一种是直接调用实例化的 Accuracy 对象,计算评测指标:
labels = np.asarray([0, 1, 2, 3])
preds = np.asarray([0, 2, 1, 3])
accuracy(preds, labels)
# {'top1': 0.5}
第二种是累积多个批次的数据后,计算评测指标:
for i in range(10):
labels = np.random.randint(0, 4, size=(100, ))
predicts = np.random.randint(0, 4, size=(100, ))
accuracy.add(predicts, labels)
accuracy.compute()
# {'top1': ...}