From 7a89a3cf7b29213ad6ed0e53f86277644be3ddc9 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 18:50:38 +0200 Subject: [PATCH] Fixed creation of final labels Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index cfc2f66..9ed0219 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -272,7 +272,7 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], continue final_image_paths.append(image_path) - final_labels.append(labels) + final_labels.extend(labels) length_dataset = len(final_image_paths) labels_np = np.array(final_labels)