Skip to content

GitLab

Commit 0b8bfa84 authored by Neal Wu's avatar Neal Wu Committed by GitHub
Browse files

Merge pull request #1055 from snnn/master 

Update download_and_convert_mnist.py for 1.0 and python3 compatibility
parents 3c50f47b c6a9fd81
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
slim/datasets/download_and_convert_mnist.py
View file @ 0b8bfa84
...@@ -125,7 +125,7 @@ def _add_to_tfrecord(data_filename, labels_filename, num_images, ...@@ -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]}) png_string = sess.run(encoded_png, feed_dict={image: images[j]})
example = dataset_utils.image_to_tfexample( 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()) tfrecord_writer.write(example.SerializeToString())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment