Skip to content
Snippets Groups Projects

Updates

Merged Tiago de Freitas Pereira requested to merge updates into master
8 files
+ 232
3
Compare changes
  • Side-by-side
  • Inline
Files
8
@@ -84,6 +84,7 @@ class TFRecord(object):
@@ -84,6 +84,7 @@ class TFRecord(object):
# Convert the image data from string back to the numbers
# Convert the image data from string back to the numbers
image = tf.decode_raw(features['train/data'], tf.float32)
image = tf.decode_raw(features['train/data'], tf.float32)
 
#image = tf.decode_raw(features['train/data'], tf.uint8)
# Cast label data into int32
# Cast label data into int32
label = tf.cast(features['train/label'], tf.int64)
label = tf.cast(features['train/label'], tf.int64)
Loading