基于两个边界框列表计算精度和召回率

computing precision and recall based on two list of bounding box

有谁知道,给定两个边界框列表,一个是预测的,另一个是真实的,是否有一些现有的python库或代码我可以用它来计算精度吗?有很多先计算 lou 的代码,我知道我可以自己计算。但我认为还应该有一些现有的工具可以用来直接获取准确度指标,这样我就能以比自己编码更聪明的方式完成我的项目。

谢谢!

是的,有很多爱from sklearn.metrics

这里是 link: http://scikit-learn.org/stable/modules/classes.html#sklearn-metrics-metrics

这是一个基本示例:

from sklearn.metrics import accuracy_score
pr_score = accuracy_score(y_test, y_pred)
print(pr_score)

其中 y_pred 是预测,y_test 是所谓的 "ground truth"

以上link您可以阅读并选择适合您的评估指标。