Added debug print for class id

Signed-off-by: Jim Martens <github@2martens.de>
2019-06-03 11:33:00 +02:00
parent 2ba17a1540
commit dc6c046da2
1 changed files with 1 additions and 0 deletions
--- a/src/twomartens/masterthesis/evaluate.py
+++ b/src/twomartens/masterthesis/evaluate.py
@ -52,6 +52,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)
            number_gt_per_class[class_id] += 1
    
    return number_gt_per_class