added python3 support to read_label_file

a86f917c · Jonas Rauber · 51238b1b · a86f917c
Commit a86f917c authored Oct 04, 2016 by Jonas Rauber
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

slim/datasets/dataset_utils.py slim/datasets/dataset_utils.py +1 -1

No files found.
--- a/slim/datasets/dataset_utils.py
+++ b/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)