Spaces:
Running
on
T4
Running
on
T4
json changed to return param
Browse files
app.py
CHANGED
@@ -187,22 +187,17 @@ def track_and_mask(vid, masked_frame, original_list, mask_list, in_fps, dt_strin
|
|
187 |
outname = (dt_string+"_output.avi")
|
188 |
print('Original Frame Count: ',len(original_list))
|
189 |
print('Mask Frame Count: ',len(mask_list))
|
190 |
-
return
|
191 |
-
mask_list: mask_list,
|
192 |
-
in_fps: in_fps,
|
193 |
-
out_video:outname}
|
194 |
|
195 |
|
196 |
-
def inpaint_video(original_frame_list, mask_list):
|
197 |
outname = (dt_string+"_result.mp4")
|
198 |
args.out_fps = in_fps
|
199 |
args.outfilename = outname
|
200 |
video_inpainting(args, original_frame_list, mask_list)
|
201 |
original_frame_list = []
|
202 |
mask_list = []
|
203 |
-
return
|
204 |
-
original_frame_list: original_frame_list,
|
205 |
-
mask_list: mask_list}
|
206 |
|
207 |
|
208 |
def get_first_frame(video):
|
@@ -269,7 +264,8 @@ with gr.Blocks() as demo:
|
|
269 |
original_image, masked_image])
|
270 |
track_mask.click(fn=track_and_mask, inputs=[
|
271 |
in_video, masked_image, original_frame_list, mask_list, in_fps, dt_string], outputs=[original_frame_list, mask_list, in_fps, out_video])
|
272 |
-
inpaint.click(fn=inpaint_video,
|
|
|
273 |
|
274 |
|
275 |
demo.launch(debug=True)
|
|
|
187 |
outname = (dt_string+"_output.avi")
|
188 |
print('Original Frame Count: ',len(original_list))
|
189 |
print('Mask Frame Count: ',len(mask_list))
|
190 |
+
return original_list, mask_list, in_fps, outname
|
|
|
|
|
|
|
191 |
|
192 |
|
193 |
+
def inpaint_video(original_frame_list, mask_list, in_fps, dt_string):
|
194 |
outname = (dt_string+"_result.mp4")
|
195 |
args.out_fps = in_fps
|
196 |
args.outfilename = outname
|
197 |
video_inpainting(args, original_frame_list, mask_list)
|
198 |
original_frame_list = []
|
199 |
mask_list = []
|
200 |
+
return outname,original_frame_list, mask_list
|
|
|
|
|
201 |
|
202 |
|
203 |
def get_first_frame(video):
|
|
|
264 |
original_image, masked_image])
|
265 |
track_mask.click(fn=track_and_mask, inputs=[
|
266 |
in_video, masked_image, original_frame_list, mask_list, in_fps, dt_string], outputs=[original_frame_list, mask_list, in_fps, out_video])
|
267 |
+
inpaint.click(fn=inpaint_video, inputs=[original_frame_list, mask_list, in_fps, dt_string],
|
268 |
+
outputs=[out_video_inpaint, original_frame_list, mask_list])
|
269 |
|
270 |
|
271 |
demo.launch(debug=True)
|