Tensorflow对象检测API中的精确调用曲线

时间:2017-09-18 08:11:11

标签: tensorflow

有没有办法在Tensorflow对象检测API中可视化Precision-Recall曲线? 我知道mAP代表曲线下面积的绝对值,但我认为实际曲线对我的应用更具代表性? 我已经在utils / metrics 1中找到了一些精度和召回值,但我不知道它们代表什么,或者更好地说,如何从它们中生成Prec / Recall曲线。 有人可以帮助我,怎么做?

1 个答案:

答案 0 :(得分:2)

compute_precision_recall输出两个长度相同的numpy数组,即。精确和召回。您可以使用matplotlib轻松绘制这些值:

import matplotlib as plt
...
precision, recall = compute_precision_recall(scores, labels, num_gt)
plt.figure()
plt.step(recall, precision, where='post' )
plt.xlabel('Recall')
plt.ylabel('Precision')
plt.xlim((0, 1))
plt.ylim((0, 1))
plt.show()
plt.close()