Przeglądaj źródła

added python3 support to read_label_file

Jonas Rauber 9 lat temu
rodzic
commit
a86f917c91
1 zmienionych plików z 1 dodań i 1 usunięć
  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)