Spaces:
Running
on
T4
Running
on
T4
output order changed
Browse files
app.py
CHANGED
|
@@ -187,10 +187,10 @@ 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 |
-
original_frame_list: original_list,
|
| 192 |
mask_list: mask_list,
|
| 193 |
-
in_fps: in_fps
|
|
|
|
| 194 |
|
| 195 |
|
| 196 |
def inpaint_video(original_frame_list, mask_list):
|
|
@@ -268,7 +268,7 @@ with gr.Blocks() as demo:
|
|
| 268 |
approve_mask.click(lambda x: [x['image'], x['mask']], first_frame, [
|
| 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=[
|
| 272 |
inpaint.click(fn=inpaint_video, outputs=[out_video_inpaint, original_frame_list, mask_list])
|
| 273 |
|
| 274 |
|
|
|
|
| 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_frame_list: original_list,
|
|
|
|
| 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):
|
|
|
|
| 268 |
approve_mask.click(lambda x: [x['image'], x['mask']], first_frame, [
|
| 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, outputs=[out_video_inpaint, original_frame_list, mask_list])
|
| 273 |
|
| 274 |
|