Browse Source

Fix inference PNG sequence output.

Peter Lin 3 years ago
parent
commit
7722b62cda
1 changed files with 3 additions and 3 deletions
  1. 3 3
      inference.py

+ 3 - 3
inference.py

@@ -101,9 +101,9 @@ def convert_video(model,
         if output_composition is not None:
             writer_com = ImageSequenceWriter(output_composition, 'png')
         if output_alpha is not None:
-            writer_pha = VideoWriter(output_alpha, 'png')
+            writer_pha = ImageSequenceWriter(output_alpha, 'png')
         if output_foreground is not None:
-            writer_fgr = VideoWriter(output_foreground, 'png')
+            writer_fgr = ImageSequenceWriter(output_foreground, 'png')
 
     # Inference
     model = model.eval()
@@ -205,4 +205,4 @@ if __name__ == '__main__':
         progress=not args.disable_progress
     )
     
-    
+