Merge pull request #1055 from snnn/master

Update download_and_convert_mnist.py for 1.0 and python3 compatibility

Merge pull request #1055 from snnn/master
Update download_and_convert_mnist.py for 1.0 and python3 compatibility
0b8bfa84 · Neal Wu · GitHub · 3c50f47b · c6a9fd81 · 0b8bfa84
Commit 0b8bfa84 authored Mar 14, 2017 by Neal Wu Committed by GitHub Mar 14, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

slim/datasets/download_and_convert_mnist.py slim/datasets/download_and_convert_mnist.py +1 -1

No files found.
--- a/slim/datasets/download_and_convert_mnist.py
+++ b/slim/datasets/download_and_convert_mnist.py
@@ -125,7 +125,7 @@ def _add_to_tfrecord(data_filename, labels_filename, num_images,
        png_string = sess.run(encoded_png, feed_dict={image: images[j]})

        example = dataset_utils.image_to_tfexample(
-            png_string, 'png', _IMAGE_SIZE, _IMAGE_SIZE, labels[j])
+            png_string, 'png'.encode(), _IMAGE_SIZE, _IMAGE_SIZE, labels[j])
        tfrecord_writer.write(example.SerializeToString())