Procházet zdrojové kódy

added python3 support to read_label_file

Jonas Rauber před 9 roky
rodič
revize
a86f917c91
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)