Added debug print for class id
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -52,6 +52,7 @@ def get_number_gt_per_class(labels: Sequence[Sequence[Sequence[int]]],
|
|||||||
# iterate over boxes in image
|
# iterate over boxes in image
|
||||||
for j in range(boxes.shape[0]):
|
for j in range(boxes.shape[0]):
|
||||||
class_id = boxes[j, 0]
|
class_id = boxes[j, 0]
|
||||||
|
print(class_id)
|
||||||
number_gt_per_class[class_id] += 1
|
number_gt_per_class[class_id] += 1
|
||||||
|
|
||||||
return number_gt_per_class
|
return number_gt_per_class
|
||||||
|
|||||||
Reference in New Issue
Block a user