From d3d57177a555eeed877fc87dcbbc041b50e81143 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 22 May 2019 15:21:11 +0200 Subject: [PATCH] Fixed indexing of tuples 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 8cbe9a9..5588132 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -348,9 +348,9 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \ return image_resized, _labels processed = tf.map_fn(_get_images, (_images, labels)) - processed_images = tf.reshape(processed[:, 0], [-1, resized_shape[0], resized_shape[1], 3]) + processed_images = tf.reshape(processed[0], [-1, resized_shape[0], resized_shape[1], 3]) - return processed_images, processed[:, 1] + return processed_images, processed[1] return _load_images_ssd