From e1488161b57792763968f8cbde417639fcde6cfe Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 22 May 2019 15:36:38 +0200 Subject: [PATCH] Fixed dtype specification (2nd) 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 7b60571..631de5b 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -347,7 +347,7 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \ return image_resized, _labels - processed = tf.map_fn(_get_images, (_images, labels), dtype=[tf.float32, tf.float32]) + processed = tf.map_fn(_get_images, (_images, labels), dtype=(tf.float32, tf.float32)) processed_images = tf.reshape(processed[0], [-1, resized_shape[0], resized_shape[1], 3]) return processed_images, processed[1]