From fb8a33cd24f4f12a2fe44348cff39297cf4b48d5 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Mon, 3 Jun 2019 11:33:59 +0200 Subject: [PATCH] Fixed indexing issue wrt the class_id Signed-off-by: Jim Martens --- src/twomartens/masterthesis/evaluate.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/evaluate.py b/src/twomartens/masterthesis/evaluate.py index d85fdef..f6925bc 100644 --- a/src/twomartens/masterthesis/evaluate.py +++ b/src/twomartens/masterthesis/evaluate.py @@ -51,8 +51,7 @@ def get_number_gt_per_class(labels: Sequence[Sequence[Sequence[int]]], # iterate over boxes in image for j in range(boxes.shape[0]): - class_id = boxes[j, 0] - print(class_id) + class_id = int(boxes[j, 0]) number_gt_per_class[class_id] += 1 return number_gt_per_class