sindhuhegde commited on
Commit
0a28502
1 Parent(s): bd4b238

Update app

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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":