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

Fixed indexing of tuples

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-05-22 15:21:11 +02:00
parent d398ecba04
commit d3d57177a5
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

@ -348,9 +348,9 @@ 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)) processed = tf.map_fn(_get_images, (_images, labels))
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]
return _load_images_ssd return _load_images_ssd