kinemotion 0.67.0__py3-none-any.whl → 0.68.0__py3-none-any.whl

kinemotion/cli.py

@@ -8,14 +8,15 @@ from .dropjump.cli import dropjump_analyze
 
 @click.group()
 @click.version_option(package_name="dropjump-analyze")
-def cli() -> None:
+def cli() -> None:  # type: ignore[return]
     """Kinemotion: Video-based kinematic analysis for athletic performance."""
     pass
 
 
 # Register commands from submodules
-cli.add_command(dropjump_analyze)
-cli.add_command(cmj_analyze)
+# Type ignore needed because @click.group() transforms cli into a click.Group
+cli.add_command(dropjump_analyze)  # type: ignore[attr-defined]
+cli.add_command(cmj_analyze)  # type: ignore[attr-defined]
 
 
 if __name__ == "__main__":

kinemotion/cmj/api.py

@@ -37,7 +37,7 @@ from ..core.pipeline_utils import (
     process_all_frames,
     process_videos_bulk_generic,
 )
-from ..core.pose import PoseTracker
+from ..core.pose import MediaPipePoseTracker
 from ..core.quality import QualityAssessment, assess_jump_quality
 from ..core.timing import PerformanceTimer, Timer
 from ..core.validation import ValidationResult
@@ -114,7 +114,7 @@ def _save_metrics_to_json(
 
 def _print_timing_summary(start_time: float, timer: Timer, metrics: CMJMetrics) -> None:
     """Print verbose timing summary and metrics."""
-    total_time = time.time() - start_time
+    total_time = time.perf_counter() - start_time
     stage_times = convert_timer_to_stage_names(timer.get_metrics())
 
     print("\n=== Timing Summary ===")
@@ -187,7 +187,7 @@ def _create_processing_info(
     start_time: float, quality_preset: QualityPreset, timer: Timer
 ) -> ProcessingInfo:
     """Create processing information metadata."""
-    processing_time = time.time() - start_time
+    processing_time = time.perf_counter() - start_time
     stage_times = convert_timer_to_stage_names(timer.get_metrics())
 
     return ProcessingInfo(
@@ -219,7 +219,8 @@ def _run_pose_tracking(
     quality_preset: QualityPreset,
     detection_confidence: float | None,
     tracking_confidence: float | None,
-    pose_tracker: PoseTracker | None,
+    pose_tracker: "MediaPipePoseTracker | None",
+    pose_backend: str | None,
     verbose: bool,
     timer: Timer,
 ) -> tuple[list[NDArray[np.uint8]], list, list[int]]:
@@ -234,15 +235,38 @@ def _run_pose_tracking(
         quality_preset, detection_confidence, tracking_confidence
     )
 
-    if verbose:
-        print("Processing all frames with MediaPipe pose tracking...")
+    if pose_tracker is None:
+        if pose_backend is not None:
+            import time
 
-    tracker = pose_tracker or PoseTracker(
-        min_detection_confidence=det_conf,
-        min_tracking_confidence=track_conf,
-        timer=timer,
-    )
-    should_close_tracker = pose_tracker is None
+            from ..core import get_tracker_info
+            from ..core.pose import PoseTrackerFactory
+
+            init_start = time.perf_counter()
+            tracker = PoseTrackerFactory.create(
+                backend=pose_backend,
+                min_detection_confidence=det_conf,
+                min_tracking_confidence=track_conf,
+                timer=timer,
+            )
+            init_time = time.perf_counter() - init_start
+
+            if verbose:
+                print(f"Using pose backend: {pose_backend}")
+                print(f"  → {get_tracker_info(tracker)}")
+                print(f"  → Initialized in {init_time * 1000:.1f} ms")
+        else:
+            if verbose:
+                print("Processing all frames with MediaPipe pose tracking...")
+            tracker = MediaPipePoseTracker(
+                min_detection_confidence=det_conf,
+                min_tracking_confidence=track_conf,
+                timer=timer,
+            )
+        should_close_tracker = True
+    else:
+        tracker = pose_tracker
+        should_close_tracker = False
 
     return process_all_frames(video, tracker, verbose, timer, close_tracker=should_close_tracker)
 
@@ -388,6 +412,7 @@ class CMJVideoConfig:
     overrides: AnalysisOverrides | None = None
     detection_confidence: float | None = None
     tracking_confidence: float | None = None
+    pose_backend: str | None = None
 
 
 @dataclass
@@ -409,9 +434,10 @@ def process_cmj_video(
     overrides: AnalysisOverrides | None = None,
     detection_confidence: float | None = None,
     tracking_confidence: float | None = None,
+    pose_backend: str | None = None,
     verbose: bool = False,
     timer: Timer | None = None,
-    pose_tracker: PoseTracker | None = None,
+    pose_tracker: MediaPipePoseTracker | None = None,
 ) -> CMJMetrics:
     """
     Process a single CMJ video and return metrics.
@@ -442,7 +468,7 @@ def process_cmj_video(
     if not Path(video_path).exists():
         raise FileNotFoundError(f"Video file not found: {video_path}")
 
-    start_time = time.time()
+    start_time = time.perf_counter()
     timer = timer or PerformanceTimer()
     quality_preset = parse_quality_preset(quality)
 
@@ -455,6 +481,7 @@ def process_cmj_video(
                 detection_confidence,
                 tracking_confidence,
                 pose_tracker,
+                pose_backend,
                 verbose,
                 timer,
             )
@@ -529,7 +556,7 @@ def process_cmj_videos_bulk(
 
 def _process_cmj_video_wrapper(config: CMJVideoConfig) -> CMJVideoResult:
     """Wrapper function for parallel CMJ processing."""
-    start_time = time.time()
+    start_time = time.perf_counter()
 
     try:
         metrics = process_cmj_video(
@@ -540,10 +567,11 @@ def _process_cmj_video_wrapper(config: CMJVideoConfig) -> CMJVideoResult:
             overrides=config.overrides,
             detection_confidence=config.detection_confidence,
             tracking_confidence=config.tracking_confidence,
+            pose_backend=config.pose_backend,
             verbose=False,
         )
 
-        processing_time = time.time() - start_time
+        processing_time = time.perf_counter() - start_time
 
         return CMJVideoResult(
             video_path=config.video_path,
@@ -553,7 +581,7 @@ def _process_cmj_video_wrapper(config: CMJVideoConfig) -> CMJVideoResult:
         )
 
     except Exception as e:
-        processing_time = time.time() - start_time
+        processing_time = time.perf_counter() - start_time
 
         return CMJVideoResult(
             video_path=config.video_path,

kinemotion/cmj/cli.py

@@ -27,6 +27,7 @@ class AnalysisParameters:
     visibility_threshold: float | None = None
     detection_confidence: float | None = None
     tracking_confidence: float | None = None
+    pose_backend: str | None = None
 
 
 def _process_batch_videos(
@@ -77,6 +78,23 @@ def _process_batch_videos(
     is_flag=True,
     help="Show auto-selected parameters and analysis details",
 )
+@click.option(
+    "--pose-backend",
+    type=click.Choice(
+        ["auto", "mediapipe", "rtmpose-cpu", "rtmpose-cuda", "rtmpose-coreml"],
+        case_sensitive=False,
+    ),
+    default="auto",
+    help=(
+        "Pose tracking backend: "
+        "auto (detect best), "
+        "mediapipe (baseline), "
+        "rtmpose-cpu (optimized CPU), "
+        "rtmpose-cuda (NVIDIA GPU), "
+        "rtmpose-coreml (Apple Silicon)"
+    ),
+    show_default=True,
+)
 # Batch processing options
 @click.option(
     "--batch",
@@ -155,6 +173,7 @@ def cmj_analyze(  # NOSONAR(S107) - Click CLI requires individual parameters
     json_output: str | None,
     quality: str,
     verbose: bool,
+    pose_backend: str,
     batch: bool,
     workers: int,
     output_dir: str | None,
@@ -218,6 +237,7 @@ def cmj_analyze(  # NOSONAR(S107) - Click CLI requires individual parameters
         visibility_threshold=visibility_threshold,
         detection_confidence=detection_confidence,
         tracking_confidence=tracking_confidence,
+        pose_backend=pose_backend,
     )
 
     if use_batch:
@@ -273,6 +293,7 @@ def _process_single(
             overrides=overrides,
             detection_confidence=expert_params.detection_confidence,
             tracking_confidence=expert_params.tracking_confidence,
+            pose_backend=expert_params.pose_backend,
             verbose=verbose,
         )
 

kinemotion/core/__init__.py

@@ -9,7 +9,12 @@ from .filtering import (
     remove_outliers,
 )
 from .model_downloader import get_model_cache_dir, get_model_path
-from .pose import PoseTracker, compute_center_of_mass
+from .pose import (
+    MediaPipePoseTracker,
+    PoseTrackerFactory,
+    compute_center_of_mass,
+    get_tracker_info,
+)
 from .pose_landmarks import KINEMOTION_LANDMARKS, LANDMARK_INDICES
 from .quality import (
     QualityAssessment,
@@ -34,8 +39,10 @@ from .video_io import VideoProcessor
 
 __all__ = [
     # Pose tracking
-    "PoseTracker",
+    "MediaPipePoseTracker",
+    "PoseTrackerFactory",
     "compute_center_of_mass",
+    "get_tracker_info",
     "LANDMARK_INDICES",
     "KINEMOTION_LANDMARKS",
     "get_model_path",

kinemotion/core/debug_overlay_utils.py

@@ -340,9 +340,9 @@ class BaseDebugOverlayRenderer:
         )
 
         try:
-            reencode_start = time.time()
+            reencode_start = time.perf_counter()
             subprocess.run(cmd, check=True, stdout=subprocess.DEVNULL, stderr=subprocess.PIPE)
-            self.reencode_duration_s = time.time() - reencode_start
+            self.reencode_duration_s = time.perf_counter() - reencode_start
 
             _log(
                 "info",

kinemotion/core/pipeline_utils.py

@@ -11,7 +11,7 @@ import numpy as np
 from ..cmj.analysis import compute_average_hip_position
 from ..dropjump.analysis import compute_average_foot_position
 from .auto_tuning import AnalysisParameters, QualityPreset, VideoCharacteristics
-from .pose import PoseTracker
+from .pose import MediaPipePoseTracker
 from .smoothing import smooth_landmarks, smooth_landmarks_advanced
 from .timing import NULL_TIMER, Timer
 from .video_io import VideoProcessor
@@ -142,7 +142,7 @@ def print_verbose_parameters(
 
 def _process_frames_loop(
     video: VideoProcessor,
-    tracker: PoseTracker,
+    tracker: MediaPipePoseTracker,
     step: int,
     should_resize: bool,
     debug_w: int,
@@ -180,7 +180,7 @@ def _process_frames_loop(
 
 def process_all_frames(
     video: VideoProcessor,
-    tracker: PoseTracker,
+    tracker: MediaPipePoseTracker,
     verbose: bool,
     timer: Timer | None = None,
     close_tracker: bool = True,