浏览代码

added python3 support to read_label_file

Jonas Rauber 9 年之前
父节点
当前提交
a86f917c91
共有 1 个文件被更改,包括 1 次插入1 次删除
  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)
   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)