kinemotion 0.70.1__py3-none-any.whl → 0.71.1__py3-none-any.whl

kinemotion/core/rtmpose_wrapper.py

@@ -1,190 +0,0 @@
-"""RTMPose wrapper for CUDA and CoreML acceleration.
-
-This adapter wraps RTMLib's BodyWithFeet for CUDA (NVIDIA GPU) and CoreML (Apple Silicon)
-acceleration, matching kinemotion's PoseTracker API.
-
-Performance:
-- CUDA (RTX 4070 Ti Super): 133 FPS (271% of MediaPipe)
-- CoreML (M1 Pro): 42 FPS (94% of MediaPipe)
-- Accuracy: 9-12px mean difference from MediaPipe
-"""
-
-from __future__ import annotations
-
-import numpy as np
-from rtmlib import BodyWithFeet
-
-from kinemotion.core.timing import NULL_TIMER, Timer
-
-# Halpe-26 to kinemotion landmark mapping
-HALPE_TO_KINEMOTION = {
-    0: "nose",
-    5: "left_shoulder",
-    6: "right_shoulder",
-    11: "left_hip",
-    12: "right_hip",
-    13: "left_knee",
-    14: "right_knee",
-    15: "left_ankle",
-    16: "right_ankle",
-    20: "left_foot_index",
-    21: "right_foot_index",
-    24: "left_heel",
-    25: "right_heel",
-}
-
-
-class RTMPoseWrapper:
-    """RTMPose wrapper for CUDA/CoreML acceleration.
-
-    Uses RTMLib's BodyWithFeet (Halpe-26 format) which provides all 13
-    kinemotion landmarks including feet.
-
-    Supports:
-    - device='cuda': NVIDIA GPU acceleration (fastest)
-    - device='mps': Apple Silicon CoreML acceleration
-    - device='cpu': Fallback (unoptimized, use rtmpose_cpu.py instead)
-
-    Attributes:
-        timer: Optional Timer for measuring operations
-        estimator: RTMLib BodyWithFeet estimator instance
-        mode: RTMLib mode ('lightweight', 'balanced', 'performance')
-        device: Target device ('cuda', 'mps', 'cpu')
-    """
-
-    def __init__(
-        self,
-        min_detection_confidence: float = 0.5,
-        min_tracking_confidence: float = 0.5,
-        timer: Timer | None = None,
-        mode: str = "lightweight",
-        backend: str = "onnxruntime",
-        device: str = "cpu",
-        pose_input_size: tuple[int, int] | None = None,
-    ) -> None:
-        """Initialize the RTMPose wrapper.
-
-        Args:
-            min_detection_confidence: Minimum confidence for pose detection
-            min_tracking_confidence: Minimum confidence for pose tracking
-            timer: Optional Timer for measuring operations
-            mode: RTMLib performance mode:
-                - 'lightweight': Fastest, RTMPose-s
-                - 'balanced': Default mode
-                - 'performance': Best accuracy, RTMPose-m
-            backend: RTMLib backend ('onnxruntime', 'opencv')
-            device: RTMLib device ('cpu', 'cuda', 'mps')
-            pose_input_size: Custom input size as (height, width) tuple
-        """
-        self.timer = timer or NULL_TIMER
-        self.mode = mode
-        self.backend = backend
-        self.device = device
-
-        with self.timer.measure("rtmpose_wrapper_initialization"):
-            kwargs = {
-                "mode": mode,
-                "backend": backend,
-                "device": device,
-            }
-            if pose_input_size is not None:
-                kwargs["pose_input_size"] = pose_input_size  # type: ignore[assignment]
-            self.estimator = BodyWithFeet(**kwargs)  # type: ignore[arg-type]
-
-    def process_frame(
-        self, frame: np.ndarray, timestamp_ms: int = 0
-    ) -> dict[str, tuple[float, float, float]] | None:
-        """Process a single frame and extract pose landmarks.
-
-        Args:
-            frame: BGR image frame (OpenCV format)
-            timestamp_ms: Frame timestamp in milliseconds (unused, for API compatibility)
-
-        Returns:
-            Dictionary mapping landmark names to (x, y, visibility) tuples,
-            or None if no pose detected. Coordinates are normalized (0-1).
-        """
-        if frame.size == 0:
-            return None
-
-        height, width = frame.shape[:2]
-
-        # RTMLib expects RGB, but BodyWithFeet handles conversion internally
-        with self.timer.measure("rtmpose_inference"):
-            keypoints, scores = self.estimator(frame)
-
-        if keypoints.shape[0] == 0:
-            return None
-
-        # Extract first person's keypoints
-        with self.timer.measure("landmark_extraction"):
-            landmarks = self._extract_landmarks(keypoints[0], scores[0], width, height)
-
-        return landmarks
-
-    def _extract_landmarks(
-        self,
-        keypoints: np.ndarray,
-        scores: np.ndarray,
-        img_width: int,
-        img_height: int,
-    ) -> dict[str, tuple[float, float, float]]:
-        """Extract and convert RTMLib landmarks to MediaPipe format.
-
-        Args:
-            keypoints: (26, 2) array of pixel coordinates
-            scores: (26,) array of confidence scores
-            img_width: Image width for normalization
-            img_height: Image height for normalization
-
-        Returns:
-            Dictionary mapping kinemotion landmark names to normalized
-            (x, y, visibility) tuples.
-        """
-        landmarks = {}
-
-        for halpe_idx, name in HALPE_TO_KINEMOTION.items():
-            x_pixel, y_pixel = keypoints[halpe_idx]
-            confidence = float(scores[halpe_idx])
-
-            # Normalize to [0, 1] like MediaPipe
-            x_norm = float(x_pixel / img_width)
-            y_norm = float(y_pixel / img_height)
-
-            # Clamp to valid range
-            x_norm = max(0.0, min(1.0, x_norm))
-            y_norm = max(0.0, min(1.0, y_norm))
-
-            # Use confidence as visibility (MediaPipe compatibility)
-            landmarks[name] = (x_norm, y_norm, confidence)
-
-        return landmarks
-
-    def close(self) -> None:
-        """Release resources (no-op for RTMLib)."""
-        pass
-
-
-def create_rtmpose_wrapper(
-    device: str = "cpu",
-    mode: str = "lightweight",
-    timer: Timer | None = None,
-) -> RTMPoseWrapper:
-    """Factory function to create an RTMPose wrapper.
-
-    Args:
-        device: Target device ('cuda', 'mps', 'cpu')
-        mode: Performance mode ('lightweight', 'balanced', 'performance')
-        timer: Optional Timer for measuring operations
-
-    Returns:
-        Configured RTMPoseWrapper instance
-
-    Example:
-        # CUDA (NVIDIA GPU)
-        tracker = create_rtmpose_wrapper(device='cuda')
-
-        # CoreML (Apple Silicon)
-        tracker = create_rtmpose_wrapper(device='mps')
-    """
-    return RTMPoseWrapper(device=device, mode=mode, timer=timer)

kinemotion-0.70.1.dist-info/RECORD

@@ -1,51 +0,0 @@
-kinemotion/__init__.py,sha256=Ho_BUtsM0PBxBW1ye9RlUg0ZqBlgGudRI9bZTF7QKUI,966
-kinemotion/api.py,sha256=uG1e4bTnj2c-6cbZJEZ_LjMwFdaG32ba2KcK_XjE_NI,1040
-kinemotion/cli.py,sha256=_Us9krSce4GUKtlLIPrFUhKmPWURzeJ1-ydR_YU2VGw,626
-kinemotion/cmj/__init__.py,sha256=SkAw9ka8Yd1Qfv9hcvk22m3EfucROzYrSNGNF5kDzho,113
-kinemotion/cmj/analysis.py,sha256=jM9ZX44h1__Cg2iIhAYRoo_5fPwIOeV5Q2FZ22rMvKY,22202
-kinemotion/cmj/api.py,sha256=Pyc0IoFyvBWcLnWq3lV9pn2ZcdFEU8ki_GX1DfXARSU,19417
-kinemotion/cmj/cli.py,sha256=-hNNN7rshrICJ7bG0EfSdEDOPNcGX_CtOZfgrZfatQg,10522
-kinemotion/cmj/debug_overlay.py,sha256=bX9aPLhXiLCCMZW9v8Y4OiOAaZO0i-UGr-Pl8HCsmbI,15810
-kinemotion/cmj/joint_angles.py,sha256=HmheIEiKcQz39cRezk4h-htorOhGNPsqKIR9RsAEKts,9960
-kinemotion/cmj/kinematics.py,sha256=KwA8uSj3g1SeNf0NXMSHsp3gIw6Gfa-6QWIwdYdRXYw,13362
-kinemotion/cmj/metrics_validator.py,sha256=3oFB331Xch2sRMTvqALiwOvsWkCUhrLQ7ZCZ4QhI2lA,30986
-kinemotion/cmj/validation_bounds.py,sha256=Ry915JdInPXbqjaVGNY_urnDO1PAkCSJqHwNKRq-VkU,12048
-kinemotion/core/__init__.py,sha256=8WB7tAJPKOxgNzbhIEOnGnkRr0CcdNeTnz91Jsiyafo,1812
-kinemotion/core/auto_tuning.py,sha256=lhAqPc-eLjMYx9BCvKdECE7TD2Dweb9KcifV6JHaXOE,11278
-kinemotion/core/cli_utils.py,sha256=sQPbT6XWWau-sm9yuN5c3eS5xNzoQGGXwSz6hQXtRvM,1859
-kinemotion/core/debug_overlay_utils.py,sha256=D4aT8xstThPcV2i5D4KJZJEttW6E_4GE5QiERqe1MwI,13049
-kinemotion/core/determinism.py,sha256=Frw-KAOvAxTL_XtxoWpXCjMbQPUKEAusK6JctlkeuRo,2509
-kinemotion/core/experimental.py,sha256=IK05AF4aZS15ke85hF3TWCqRIXU1AlD_XKzFz735Ua8,3640
-kinemotion/core/filtering.py,sha256=Oc__pV6iHEGyyovbqa5SUi-6v8QyvaRVwA0LRayM884,11355
-kinemotion/core/formatting.py,sha256=G_3eqgOtym9RFOZVEwCxye4A2cyrmgvtQ214vIshowU,2480
-kinemotion/core/metadata.py,sha256=bJAVa4nym__zx1hNowSZduMGKBSGOPxTbBQkjm6N0D0,7207
-kinemotion/core/model_downloader.py,sha256=mqhJBHGaNe0aN9qbcBqvcTk9FDd7xaHqEcwD-fyP89c,5205
-kinemotion/core/pipeline_utils.py,sha256=B5jMXoiLaTh02uGA2MIe1uZLVSRGZ5nxbARuvdrjDrQ,15161
-kinemotion/core/pose.py,sha256=d5hCWiKifuB80shOC68uuRIYoXVBOQOtU54RjgNaTqc,28637
-kinemotion/core/pose_landmarks.py,sha256=LcEbL5K5xKia6dCzWf6Ft18UIE1CLMMqCZ3KUjwUDzM,1558
-kinemotion/core/quality.py,sha256=VUkRL2N6B7lfIZ2pE9han_U68JwarmZz1U0ygHkgkhE,13022
-kinemotion/core/rtmpose_cpu.py,sha256=Mox8Hon3hulyA6uHKUIe2hCR4xinDZfotOOIVncXi2M,22356
-kinemotion/core/rtmpose_wrapper.py,sha256=R5QlZIHJczM1AxvVqF256DcD_LvGvHyjrFnxCsWc7do,6202
-kinemotion/core/smoothing.py,sha256=ELMHL7pzSqYffjnLDBUMBJIgt1AwOssDInE8IiXBbig,15942
-kinemotion/core/timing.py,sha256=ITX77q4hbtajRuWfgwYhws8nCvOeKFlEdKjCu8lD9_w,7938
-kinemotion/core/types.py,sha256=A_HclzKpf3By5DiJ0wY9B-dQJrIVAAhUfGab7qTSIL8,1279
-kinemotion/core/validation.py,sha256=0xVv-ftWveV60fJ97kmZMuy2Qqqb5aZLR50dDIrjnhg,6773
-kinemotion/core/video_io.py,sha256=TxdLUEpekGytesL3X3k78WWgZTOd5fuge30hU4Uy48Y,9198
-kinemotion/dropjump/__init__.py,sha256=tC3H3BrCg8Oj-db-Vrtx4PH_llR1Ppkd5jwaOjhQcLg,862
-kinemotion/dropjump/analysis.py,sha256=YomuoJF_peyrBSpeT89Q5_sBgY0kEDyq7TFrtEnRLjs,28049
-kinemotion/dropjump/api.py,sha256=0AGA896H97x9GZoO8kX75iBCNCHEEm1lH99yoqJceW8,21522
-kinemotion/dropjump/cli.py,sha256=sJccY6HrRqj6DKERtsUgy__Wmp-pdm5c765EN3ydiec,17562
-kinemotion/dropjump/debug_overlay.py,sha256=9RQYXPRf0q2wdy6y2Ak2R4tpRceDwC8aJrXZzkmh3Wo,5942
-kinemotion/dropjump/kinematics.py,sha256=dx4PuXKfKMKcsc_HX6sXj8rHXf9ksiZIOAIkJ4vBlY4,19637
-kinemotion/dropjump/metrics_validator.py,sha256=lSfo4Lm5FHccl8ijUP6SA-kcSh50LS9hF8UIyWxcnW8,9243
-kinemotion/dropjump/validation_bounds.py,sha256=x4yjcFxyvdMp5e7MkcoUosGLeGsxBh1Lft6h__AQ2G8,5124
-kinemotion/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kinemotion/models/pose_landmarker_lite.task,sha256=WZKeHR7pUodzXd2DOxnPSsRtKbx6_du_Z1PEWWkNV0o,5777746
-kinemotion/models/rtmpose-s_simcc-body7_pt-body7-halpe26_700e-256x192-7f134165_20230605.onnx,sha256=hcXWaaoLDHlpeZ6p4GINO4wdPCtoeCQ-mKHvg1FFeYY,22793379
-kinemotion/models/yolox_tiny_8xb8-300e_humanart-6f3252f9.onnx,sha256=zrEcBymPlcUNfFq-uQbQM0DIXyOqeePmaWbn-2wwclA,20283006
-kinemotion/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kinemotion-0.70.1.dist-info/METADATA,sha256=nQ1AOdsgyVvO96oAwTXuAJ7Gbi29FYbg2mv6UCsF7h0,26372
-kinemotion-0.70.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-kinemotion-0.70.1.dist-info/entry_points.txt,sha256=zaqnAnjLvcdrk1Qvj5nvXZCZ2gp0prS7it1zTJygcIY,50
-kinemotion-0.70.1.dist-info/licenses/LICENSE,sha256=KZajvqsHw0NoOHOi2q0FZ4NBe9HdV6oey-IPYAtHXfg,1088
-kinemotion-0.70.1.dist-info/RECORD,,