Skip to content
Snippets Groups Projects

Implemented random triplet batching

Merged Tiago de Freitas Pereira requested to merge issue_43 into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -62,7 +62,7 @@ def create_mnist_tfrecord(tfrecords_filename, data, labels, n_samples=6000):
@@ -62,7 +62,7 @@ def create_mnist_tfrecord(tfrecords_filename, data, labels, n_samples=6000):
img_raw = img.tostring()
img_raw = img.tostring()
feature = {'data': _bytes_feature(img_raw),
feature = {'data': _bytes_feature(img_raw),
'label': _int64_feature(labels[i]),
'label': _int64_feature(labels[i]),
'key': _bytes_feature("")
'key': _bytes_feature(b'-')
}
}
example = tf.train.Example(features=tf.train.Features(feature=feature))
example = tf.train.Example(features=tf.train.Features(feature=feature))
Loading