Skip to content
Snippets Groups Projects

Updates

Merged Tiago de Freitas Pereira requested to merge updates into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -152,7 +152,7 @@ def test_cnn_tfrecord():
@@ -152,7 +152,7 @@ def test_cnn_tfrecord():
img = data[i]
img = data[i]
img_raw = img.tostring()
img_raw = img.tostring()
feature = {'train/image': _bytes_feature(img_raw),
feature = {'train/data': _bytes_feature(img_raw),
'train/label': _int64_feature(labels[i])
'train/label': _int64_feature(labels[i])
}
}
Loading