From f5db36bf9f5388e59143b4391a06d4c7f6cbb87d Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Sun, 28 Apr 2019 18:44:25 +0200 Subject: [PATCH] Attempt conversion to Numpy array before converting to Tensor 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 e91883e..bcccd70 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -277,7 +277,7 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]], length_dataset = len(final_image_paths) path_dataset = tf.data.Dataset.from_tensor_slices(final_image_paths) - label_dataset = tf.data.Dataset.from_tensor_slices(final_labels) + label_dataset = tf.data.Dataset.from_tensor_slices(np.ndarray(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)