diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index a9881f4..e91883e 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -260,13 +260,13 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], labels = [] for instance in frame_instances: bbox = instance['bbox'] - labels.append(( + labels.append([ cats_to_classes[instance['coco_id']], bbox[0], bbox[1], bbox[2], bbox[3] - )) + ]) if not labels: continue