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

Fixed dtype specification

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-28 19:12:46 +02:00
parent dbab9e36b5
commit 36b11aaa5e
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

@ -316,7 +316,7 @@ def _load_images_ssd_callback(resized_shape: Sequence[int]) \
return image_resized return image_resized
processed = tf.map_fn(_get_images, _images, dtype=[tf.float32, tf.float32]) processed = tf.map_fn(_get_images, _images, dtype=tf.float32)
processed_images = tf.reshape(processed, [-1, resized_shape[0], resized_shape[1], 3]) processed_images = tf.reshape(processed, [-1, resized_shape[0], resized_shape[1], 3])
return processed_images return processed_images