Spaces:
Running
on
Zero
Running
on
Zero
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":
|