From ba1b3d8fe5c509e0761be36806b9b1925418fb49 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 10 Sep 2018 16:19:00 +0200 Subject: [PATCH] updates --- utils/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/utils.py b/utils/utils.py index 121f352e..bccb4278 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -99,7 +99,7 @@ def ap_per_class(tp, conf, pred_cls, target_cls): tp, conf, pred_cls = tp[i], conf[i], pred_cls[i] # Find unique classes - unique_classes = np.unique(np.concatenate((pred_cls, target_cls), 0)) + unique_classes = target_cls #np.unique(np.concatenate((pred_cls, target_cls), 0)) # Create Precision-Recall curve and compute AP for each class ap = []