Skip to content

GitLab

Commit 77080ad1 authored by Asim Shankar's avatar Asim Shankar Committed by GitHub
Browse files

Merge pull request #503 from jonasrauber/patch 

added python3 support to read_label_file
parents 1fc268ca a86f917c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
slim/datasets/dataset_utils.py
View file @ 77080ad1
...@@ -125,7 +125,7 @@ def read_label_file(dataset_dir, filename=LABELS_FILENAME): ...@@ -125,7 +125,7 @@ def read_label_file(dataset_dir, filename=LABELS_FILENAME):
""" """
labels_filename = os.path.join(dataset_dir, filename) labels_filename = os.path.join(dataset_dir, filename)
with tf.gfile.Open(labels_filename, 'r') as f: with tf.gfile.Open(labels_filename, 'r') as f:
lines = f.read() lines = f.read().decode()
lines = lines.split('\n') lines = lines.split('\n')
lines = filter(None, lines) lines = filter(None, lines)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment