Pārlūkot izejas kodu

Merge pull request #1055 from snnn/master

Update download_and_convert_mnist.py for 1.0 and python3 compatibility
Neal Wu 8 gadi atpakaļ
vecāks
revīzija
0b8bfa849c
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      slim/datasets/download_and_convert_mnist.py

+ 1 - 1
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())