Colin Leong commited on
Commit
e064361
·
1 Parent(s): 77d65a9

fix crash on youtube-asl filtering

Browse files
Files changed (1) hide show
  1. app.py +14 -3
app.py CHANGED
@@ -182,7 +182,7 @@ if uploaded_file is not None:
182
  # These are indices 0, 4, 13, 14, 17, 33, 37, 39, 46, 52, 55, 61, 64, 81, 82, 93, 133, 151, 152, 159, 172, 178,
183
  # 181, 263, 269, 276, 282, 285, 291, 294, 311, 323, 362, 386, 397, 468, 473.
184
  # Colin: note that these are with refine_face_landmarks on, and are relative to the component itself. Working it all out the result is:
185
- components=['POSE_LANDMARKS', 'FACE_LANDMARKS', 'LEFT_HAND_LANDMARKS', 'RIGHT_HAND_LANDMARKS']
186
  points_dict={
187
  "POSE_LANDMARKS": [
188
  "LEFT_SHOULDER",
@@ -228,11 +228,22 @@ if uploaded_file is not None:
228
  "362",
229
  "386",
230
  "397",
231
- "468", # 468 only exists with the refine_face_landmarks option on MediaPipe
232
- "473", # 473 only exists with the refine_face_landmarks option on MediaPipe
233
  ]
234
  }
235
 
 
 
 
 
 
 
 
 
 
 
 
 
 
236
  # Filter button logic
237
  # Filter section
238
  st.write("### Filter .pose File")
 
182
  # These are indices 0, 4, 13, 14, 17, 33, 37, 39, 46, 52, 55, 61, 64, 81, 82, 93, 133, 151, 152, 159, 172, 178,
183
  # 181, 263, 269, 276, 282, 285, 291, 294, 311, 323, 362, 386, 397, 468, 473.
184
  # Colin: note that these are with refine_face_landmarks on, and are relative to the component itself. Working it all out the result is:
185
+ chosen_component_names=['POSE_LANDMARKS', 'FACE_LANDMARKS', 'LEFT_HAND_LANDMARKS', 'RIGHT_HAND_LANDMARKS']
186
  points_dict={
187
  "POSE_LANDMARKS": [
188
  "LEFT_SHOULDER",
 
228
  "362",
229
  "386",
230
  "397",
 
 
231
  ]
232
  }
233
 
234
+ # check if we have the extra points from refine_face_landmarks
235
+ additional_face_points = ["468", "473"]
236
+ for additional_point in additional_face_points:
237
+ try:
238
+ point_index = pose.header.get_point_index("FACE_LANDMARKS", additional_point)
239
+ points_dict['FACE_LANDMARKS'].append(additional_point)
240
+ except ValueError:
241
+ # not in the list
242
+ # st.write(f"Point {additional_point} not in file")
243
+ pass
244
+
245
+
246
+
247
  # Filter button logic
248
  # Filter section
249
  st.write("### Filter .pose File")