Spaces:
Paused
Paused
Update inference_propainter.py
Browse files- inference_propainter.py +2 -2
inference_propainter.py
CHANGED
@@ -468,8 +468,8 @@ if __name__ == '__main__':
|
|
468 |
# save videos frame
|
469 |
masked_frame_for_save = [cv2.resize(f, out_size) for f in masked_frame_for_save]
|
470 |
comp_frames = [cv2.resize(f, out_size) for f in comp_frames]
|
471 |
-
imageio.mimwrite(os.path.join(save_root, 'masked_in.mp4'), masked_frame_for_save, fps=fps, quality=
|
472 |
-
imageio.mimwrite(os.path.join(save_root, 'inpaint_out.mp4'), comp_frames, fps=fps, quality=
|
473 |
|
474 |
print(f'\nAll results are saved in {save_root}')
|
475 |
|
|
|
468 |
# save videos frame
|
469 |
masked_frame_for_save = [cv2.resize(f, out_size) for f in masked_frame_for_save]
|
470 |
comp_frames = [cv2.resize(f, out_size) for f in comp_frames]
|
471 |
+
imageio.mimwrite(os.path.join(save_root, 'masked_in.mp4'), masked_frame_for_save, fps=fps, quality=10)
|
472 |
+
imageio.mimwrite(os.path.join(save_root, 'inpaint_out.mp4'), comp_frames, fps=fps, quality=10)
|
473 |
|
474 |
print(f'\nAll results are saved in {save_root}')
|
475 |
|