goryhon commited on
Commit
3e88aa5
·
verified ·
1 Parent(s): 1420a59

Update inference_propainter.py

Browse files
Files changed (1) hide show
  1. 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=7)
472
- imageio.mimwrite(os.path.join(save_root, 'inpaint_out.mp4'), comp_frames, fps=fps, quality=7)
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