Przeglądaj źródła

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

Kentaro Wada 7 lat temu
rodzic
commit
9f2de0534f
1 zmienionych plików z 3 dodań i 2 usunięć
  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]