Ver código fonte

Fix inference PNG sequence output.

Peter Lin 3 anos atrás
pai
commit
7722b62cda
1 arquivos alterados com 3 adições e 3 exclusões
  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
     )
     
-    
+