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

Fixed dtype specification (2nd)

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-22 15:36:38 +02:00
parent 62a65bcccb
commit e1488161b5
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 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]) processed_images = tf.reshape(processed[0], [-1, resized_shape[0], resized_shape[1], 3])
return processed_images, processed[1] return processed_images, processed[1]