Fix inference PNG sequence output.

pull/56/head
Peter Lin 3 years ago committed by GitHub
parent 1af31d1950
commit 7722b62cda
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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
)

Loading…
Cancel
Save