From 497e8c911e2dfbd2c312d0d85ff130ff91b6ae95 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 18:40:25 +0200 Subject: [PATCH] Converted tuple to list for list of labels Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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