per-class mAP report

2018-10-10 17:07:21 +02:00
parent f79e7ffa76
commit d336e0053d
3 changed files with 19 additions and 3 deletions
@@ -130,7 +130,7 @@ def ap_per_class(tp, conf, pred_cls, target_cls):
            # AP from recall-precision curve
            ap.append(compute_ap(recall, precision))

-    return np.array(ap)
+    return np.array(ap), unique_classes.astype('int32')


 def compute_ap(recall, precision):