fix bug

gnes-ai · Aug 23, 2019 · 02f70a0 · 02f70a0
1 parent 9d45567
commit 02f70a0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gnes/preprocessor/io_utils/video.py b/gnes/preprocessor/io_utils/video.py
@@ -59,7 +59,8 @@ def capture_frames(filename: str = 'pipe:',
     out, err = stream.run(
         input=video_data, capture_stdout=True, capture_stderr=True)
 
-    # video_width, video_height = extract_frame_size(err.decode())
+    if not scale:
+        width, height = extract_frame_size(err.decode())
 
     depth = 3
     if pix_fmt == 'rgba':