fix: processing

build_exe.nu

@@ -2,4 +2,6 @@
 --include-plugin-directory=.venv/Lib/site-packages/pygame_gui/data/
 --include-data-files=.venv/Lib/site-packages/pygame_gui/data/*.*=pygame_gui/data/
 --include-data-files=.venv/Lib/site-packages/pygame_gui/data/translations/*=pygame_gui/data/translations/
+--enable-console
+--enable-plugin=no-qt
 --onefile ./drumpy/cli.py)

drumpy/app/main.py

@@ -105,11 +105,11 @@ def start(self: Self) -> None:
 
 def main() -> None:
     app = App(
-        source=Source.CAMERA,
-        running_mode=RunningMode.LIVE_STREAM,  # type: ignore
-        model=LandmarkerModel.FULL,
+        source=Source.FILE,
+        running_mode=RunningMode.VIDEO,  # type: ignore
+        model=LandmarkerModel.LITE,
         delegate=BaseOptions.Delegate.CPU,  # type: ignore
-        landmark_type=LandmarkType.WORLD_LANDMARKS,
+        landmark_type=LandmarkType.LANDMARKS,
         file_path="../data/Recordings/multicam_asil_01_front.mkv",
         # log_file="test.csv",
         disable_drum=False,

drumpy/mediapipe_pose/process_result.py

@@ -42,16 +42,18 @@ def mollifier(self, x: float) -> float:
         if x <= 0:
             return 0
 
-        return math.exp((-(math.log(x / self.peak) ** 4)) * self.tightness)
+        return math.exp((-(math.log(x / self.peak) ** 2)) * self.tightness)
 
     def process_result(
         self, result: PoseLandmarkerResult, timestamp_ms: float
     ) -> PoseLandmarkerResult:
         """
         Process the result of the pose estimation
         """
-        if len(self.results) == 0 or len(result.pose_landmarks) == 0:
+        if result.pose_landmarks is None or len(result.pose_landmarks) == 0:
+            print("No pose landmarks found")
             return result
+
         for i, landmark in enumerate(result.pose_landmarks[0]):
             result.pose_landmarks[0][i] = self.process_landmark(
                 landmark, i, timestamp_ms
@@ -169,9 +171,3 @@ def calculate_diff(
         diff.y = current.y - previous.y
         diff.z = current.z - previous.z
         return diff
-
-
-if __name__ == "__main__":
-    processor = ResultProcessor(LandmarkType.LANDMARKS)
-    for i in range(20):
-        print(processor.mollifier(i / 2))