Fix ascii codec issue when outputing pascal file

HumanSignal · Jan 29, 2017 · 1fb28d8 · 1fb28d8
1 parent b7913f4
commit 1fb28d8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/libs/pascal_voc_io.py b/libs/pascal_voc_io.py
@@ -5,6 +5,7 @@
 from xml.etree import ElementTree
 from xml.etree.ElementTree import Element, SubElement
 from lxml import etree
+import codecs
 
 
 class PascalVocWriter:
@@ -100,9 +101,9 @@ def save(self, targetFile=None):
         self.appendObjects(root)
         out_file = None
         if targetFile is None:
-            out_file = open(self.filename + '.xml', 'w')
+            out_file = codecs.open(self.filename + '.xml', 'w', encoding='utf-8')
         else:
-            out_file = open(targetFile, 'w')
+            out_file = codecs.open(targetFile, 'w', encoding='utf-8')
 
         prettifyResult = self.prettify(root)
         out_file.write(prettifyResult.decode('utf8'))