fps param added to vid output

FGT_codes/tool/video_inpainting.py

@@ -776,7 +776,7 @@ def video_inpainting(args, imgArr, imgMaskArr):
     for i in range(len(comp_frames)):
         comp_frames[i] = comp_frames[i].astype(np.uint8)
     imageio.mimwrite(
-        os.path.join(args.outroot, "result.mp4"), comp_frames, fps=30, quality=8
+        os.path.join(args.outroot, "result.mp4"), comp_frames, fps=args.out_fps, quality=8
     )
     print(f"Done, please check your result in {args.outroot} ")
 

app.py

@@ -51,6 +51,7 @@ object_x = 0
 object_y = 0
 object_width = 0
 object_height = 0
+in_fps = 24
 original_frame_list = []
 mask_list = []
 
@@ -113,6 +114,8 @@ parser.add_argument('--step', type=int, default=10)
 parser.add_argument('--num_ref', type=int, default=-1)
 parser.add_argument('--neighbor_stride', type=int, default=5)
 
+parser.add_argument('--out_fps', type=int, default=24)
+
 # visualization
 parser.add_argument('--vis_flows', action='store_true',
                     help='Visualize the initialized flows')
@@ -150,6 +153,7 @@ def track_and_mask(vid, original_frame, masked_frame):
             video_capture.get(cv2.CAP_PROP_FRAME_HEIGHT))
         fps = video_capture.get(cv2.CAP_PROP_FPS)
 
+        in_fps = fps
         # can't write out mp4, so try to write into an AVI file
         video_writer = cv2.VideoWriter(
             "output.avi", cv2.VideoWriter_fourcc(*'MP42'), fps, (width, height))
@@ -203,6 +207,7 @@ def track_and_mask(vid, original_frame, masked_frame):
 
 
 def inpaint_video():
+    args.out_fps = in_fps
     video_inpainting(args, original_frame_list, mask_list)
 
     return "result.mp4"