Fingerspelling embedding + ClearML

normalization/blazepose_mapping.py

@@ -61,20 +61,25 @@ def map_blazepose_keypoint(column):
     return f"{mapped}_{hand}{suffix}"
 
 
-def map_blazepose_df(df):
+def map_blazepose_df(df, rename=True):
+    to_drop = []
+    if rename:
+        renamings = {}
+        for column in df.columns:
+            mapped_column = map_blazepose_keypoint(column)
+            if mapped_column:
+                renamings[column] = mapped_column
+            else:
+                to_drop.append(column)
+        df = df.rename(columns=renamings)
+
     for index, row in df.iterrows():
 
+        sequence_size = len(row["leftEar_Y"])
         lsx = row["leftShoulder_X"]
         rsx = row["rightShoulder_X"]
         lsy = row["leftShoulder_Y"]
         rsy = row["rightShoulder_Y"]
-        # convert all to list
-        lsx = lsx[1:-1].split(",")
-        rsx = rsx[1:-1].split(",")
-        lsy = lsy[1:-1].split(",")
-        rsy = rsy[1:-1].split(",")
-        sequence_size = len(lsx)
-
         neck_x = []
         neck_y = []
         # Treat each element of the sequence (analyzed frame) individually
@@ -84,4 +89,5 @@ def map_blazepose_df(df):
         df.loc[index, "neck_X"] = str(neck_x)
         df.loc[index, "neck_Y"] = str(neck_y)
 
-    return df
+    df.drop(columns=to_drop, inplace=True)
+    return df