Quellcode durchsuchen

Fix inference PNG sequence output.

Peter Lin vor 3 Jahren
Ursprung
Commit
7722b62cda
1 geänderte Dateien mit 3 neuen und 3 gelöschten Zeilen
  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
     )
     
-    
+