fix some bugs

EnableTech · May 22, 2018 · 562b4a9 · 562b4a9
1 parent 3f93a89
commit 562b4a9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/convert.py b/convert.py
@@ -37,7 +37,7 @@ def unique_config_sections(config_file):
     Adds unique suffixes to config sections for compability with configparser.
     """
     section_counters = defaultdict(int)
-    output_stream = io.StringIO()
+    output_stream = io.BytesIO()
     with open(config_file) as fin:
         for line in fin:
             if line.startswith('['):

diff --git a/yolo.py b/yolo.py
@@ -177,7 +177,7 @@ def detect_video(yolo, video_path):
 
 def detect_img(yolo):
     while True:
-        img = input('Input image filename:')
+        img = raw_input('Input image filename:')
         try:
             image = Image.open(img)
         except: