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

Guaranteed float dtype for image and label data

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-22 15:30:37 +02:00
parent d3d57177a5
commit 0b6e40878c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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))
processed = tf.map_fn(_get_images, (_images, labels), dtype=tf.float32)
processed_images = tf.reshape(processed[0], [-1, resized_shape[0], resized_shape[1], 3])
return processed_images, processed[1]