Procházet zdrojové kódy

Merge pull request #503 from jonasrauber/patch

added python3 support to read_label_file
Asim Shankar před 9 roky
rodič
revize
77080ad11f
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      slim/datasets/dataset_utils.py

+ 1 - 1
slim/datasets/dataset_utils.py

@@ -125,7 +125,7 @@ def read_label_file(dataset_dir, filename=LABELS_FILENAME):
   """
   labels_filename = os.path.join(dataset_dir, filename)
   with tf.gfile.Open(labels_filename, 'r') as f:
-    lines = f.read()
+    lines = f.read().decode()
   lines = lines.split('\n')
   lines = filter(None, lines)