From ff04315f96b9d1433d4075388ed79752f36eafd7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 10 Sep 2018 16:35: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 fe620193..fc590bc1 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 = []