2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fixed missing argument

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-22 14:44:02 +02:00
parent dbf243c731
commit 42d360991c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/twomartens/masterthesis/data.py
View File

@ -331,8 +331,8 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \
""" """
_images = tf.map_fn(lambda path: tf.read_file(path), paths) _images = tf.map_fn(lambda path: tf.read_file(path), paths)
def _get_images(image_data: tf.Tensor, def _get_images(data: tf.Tensor) -> Tuple[tf.Tensor, Sequence[int]]:
_labels: Sequence[int]) -> Tuple[tf.Tensor, Sequence[int]]: image_data, _labels = data
image = tf.image.decode_image(image_data, channels=3, dtype=tf.float32) image = tf.image.decode_image(image_data, channels=3, dtype=tf.float32)
image_shape = tf.shape(image) image_shape = tf.shape(image)
_labels[:, 5] = tf.cast(image_shape[0], dtype=tf.float32) / resized_shape[0] _labels[:, 5] = tf.cast(image_shape[0], dtype=tf.float32) / resized_shape[0]