2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed indexing issue wrt the class_id

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-06-03 11:33:59 +02:00
parent dc6c046da2
commit fb8a33cd24
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/twomartens/masterthesis/evaluate.py
View File

@ -51,8 +51,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 = int(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