在yolo目标检测的评价指标中有如下字段:
指标的计算过程如图:
计算过程从前到后,涉及到多个指标的计算,分别是:
- 模型输出值
- IOU计算
- 预测结果判断
- 混淆矩阵计算
- PR以及PR曲线计算
- AP计算
- mAP计算
- ROC计算
下面详细介绍各个步骤的涉及的概念,计算方法。
模型输出
在目标检测中,一张图像推理可能会输出多个预测框,每一个预测框有三个值:
- 预测框坐标
- 置信度
- 类别
如下输出分别代表:x y w h confidence cls
array([[ 1166, 712, 1273, 991, 0.94275, 1],
[ 1267, 711, 1359, 968, 0.93491, 1],
[ 302, 534, 337, 655, 0.86046, 1],
[ 339, 520, 378, 638, 0.85005, 1],
[ 382,