Spaces:
Running
on
Zero
Running
on
Zero
sindhuhegde
commited on
Commit
•
0a28502
1
Parent(s):
bd4b238
Update app
Browse files
app.py
CHANGED
@@ -365,8 +365,9 @@ def load_rgb_masked_frames(input_frames, kp_dict, stride=1, window_frames=25, wi
|
|
365 |
|
366 |
input_keypoints, resolution = kp_dict['kps'], kp_dict['resolution']
|
367 |
|
|
|
368 |
input_frames_masked = []
|
369 |
-
for i, frame_kp_dict in enumerate(input_keypoints):
|
370 |
|
371 |
img = input_frames[i]
|
372 |
face = frame_kp_dict["face"]
|
@@ -697,7 +698,7 @@ def process_video(video_path, num_avg_frames):
|
|
697 |
kp_dict, status = get_keypoints(frames)
|
698 |
if status != "success":
|
699 |
return status, None
|
700 |
-
print("Successfully extracted the keypoints")
|
701 |
|
702 |
status = check_visible_gestures(kp_dict)
|
703 |
if status != "success":
|
|
|
365 |
|
366 |
input_keypoints, resolution = kp_dict['kps'], kp_dict['resolution']
|
367 |
|
368 |
+
print("Creating masked input frames...")
|
369 |
input_frames_masked = []
|
370 |
+
for i, frame_kp_dict in tqdm(enumerate(input_keypoints)):
|
371 |
|
372 |
img = input_frames[i]
|
373 |
face = frame_kp_dict["face"]
|
|
|
698 |
kp_dict, status = get_keypoints(frames)
|
699 |
if status != "success":
|
700 |
return status, None
|
701 |
+
print("Successfully extracted the keypoints: ", len(kp_dict), len(kp_dict["kps"]))
|
702 |
|
703 |
status = check_visible_gestures(kp_dict)
|
704 |
if status != "success":
|