Update inference_propainter.py

inference_propainter.py

@@ -212,12 +212,12 @@ if __name__ == '__main__':
     parser.add_argument(
         '--save_frames', action='store_true', help='Save output frames. Default: False')
     parser.add_argument(
-        '--fp32', action='store_true', help='Use fp16 (half precision) during inference. Default: fp32 (single precision).')
+        '--fp16', action='store_true', help='Use fp16 (half precision) during inference. Default: fp32 (single precision).')
 
     args = parser.parse_args()
 
     # Use fp16 precision during inference to reduce running memory cost
-    use_half = False
+    use_half = True if args.fp16 else False
     if device == torch.device('cpu'):
         use_half = False
 
@@ -468,8 +468,8 @@ if __name__ == '__main__':
     # save videos frame
     masked_frame_for_save = [cv2.resize(f, out_size) for f in masked_frame_for_save]
     comp_frames = [cv2.resize(f, out_size) for f in comp_frames]
-    imageio.mimwrite(os.path.join(save_root, 'masked_in.mp4'), masked_frame_for_save, fps=fps, quality=10)
-    imageio.mimwrite(os.path.join(save_root, 'inpaint_out.mp4'), comp_frames, fps=fps, quality=10)
+    imageio.mimwrite(os.path.join(save_root, 'masked_in.mp4'), masked_frame_for_save, fps=fps, quality=7)
+    imageio.mimwrite(os.path.join(save_root, 'inpaint_out.mp4'), comp_frames, fps=fps, quality=7)
     
     print(f'\nAll results are saved in {save_root}')