From d12e7ec30d1579a124672b97dbc771e262317393 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 18:47:51 +0200 Subject: [PATCH] Changed code to get shape of labels Signed-off-by: Jim Martens --- src/twomartens/masterthesis/data.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/data.py b/src/twomartens/masterthesis/data.py index ea67826..cfc2f66 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -275,9 +275,11 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], final_labels.append(labels) length_dataset = len(final_image_paths) + labels_np = np.array(final_labels) + print(labels_np.shape) path_dataset = tf.data.Dataset.from_tensor_slices(final_image_paths) - label_dataset = tf.data.Dataset.from_tensor_slices(np.array(final_labels)) + label_dataset = tf.data.Dataset.from_tensor_slices(final_labels) dataset = tf.data.Dataset.zip((path_dataset, label_dataset)) dataset = dataset.apply(tf.data.experimental.shuffle_and_repeat(buffer_size=length_dataset, count=num_epochs)) dataset = dataset.batch(batch_size=batch_size)