From 42d360991c3c518446dc09a20b046f830e358a64 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 22 May 2019 14:44:02 +0200 Subject: [PATCH] Fixed missing argument 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 8a239f7..f3f91f2 100644 --- a/src/twomartens/masterthesis/data.py +++ b/src/twomartens/masterthesis/data.py @@ -331,8 +331,8 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \ """ _images = tf.map_fn(lambda path: tf.read_file(path), paths) - def _get_images(image_data: tf.Tensor, - _labels: Sequence[int]) -> Tuple[tf.Tensor, Sequence[int]]: + def _get_images(data: tf.Tensor) -> Tuple[tf.Tensor, Sequence[int]]: + image_data, _labels = data image = tf.image.decode_image(image_data, channels=3, dtype=tf.float32) image_shape = tf.shape(image) _labels[:, 5] = tf.cast(image_shape[0], dtype=tf.float32) / resized_shape[0]