apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix ascii codec issue when outputing pascal file

Browse Source
This commit is contained in:
tzutalin
2017-01-29 22:51:44 +08:00
parent b7913f4de9
commit 1fb28d81a7
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/pascal_voc_io.py
+3 -2
View File
@@ -5,6 +5,7 @@ import sys
from xml.etree import ElementTree from xml.etree import ElementTree
from xml.etree.ElementTree import Element, SubElement from xml.etree.ElementTree import Element, SubElement
from lxml import etree from lxml import etree
import codecs
class PascalVocWriter: class PascalVocWriter:
@@ -100,9 +101,9 @@ class PascalVocWriter:
self.appendObjects(root) self.appendObjects(root)
out_file = None out_file = None
if targetFile is None: if targetFile is None:
out_file = open(self.filename + '.xml', 'w') out_file = codecs.open(self.filename + '.xml', 'w', encoding='utf-8')
else: else:
out_file = open(targetFile, 'w') out_file = codecs.open(targetFile, 'w', encoding='utf-8')
prettifyResult = self.prettify(root) prettifyResult = self.prettify(root)
out_file.write(prettifyResult.decode('utf8')) out_file.write(prettifyResult.decode('utf8'))