浏览代码

Use codecs.open(encoding='utf-8') to support CJK label list file

Kentaro Wada 7 年之前
父节点
当前提交
9f2de0534f
共有 1 个文件被更改,包括 3 次插入2 次删除
  1. 3 2
      labelme/app.py

+ 3 - 2
labelme/app.py

@@ -1,4 +1,5 @@
 import argparse
+import codecs
 import functools
 import os.path
 import re
@@ -1231,8 +1232,8 @@ def main():
             sys.exit(1)
     else:
         if os.path.isfile(args.labels):
-            args.labels = [l.strip() for l in open(args.labels, 'r')
-                           if l.strip()]
+            with codecs.open(args.labels, 'r', encoding='utf-8') as f:
+                args.labels = [l.strip() for l in f if l.strip()]
         else:
             args.labels = [l for l in args.labels.split(',') if l]