feat(v1): compose temporal fall event pipeline

This commit is contained in:
ila
2026-07-21 11:57:22 +08:00
parent 906e06565e
commit f03662bf5a
16 changed files with 554 additions and 34 deletions
+139
View File
@@ -0,0 +1,139 @@
"""Compose Pose, tracking, evidence policy, and temporal fall state."""
from dataclasses import dataclass
from typing import Dict, Sequence, Tuple
from v1.config import AppConfig
from v1.evidence import PoseEvidence, assess_pose_quality, extract_evidence
from v1.fall_policy import FallEvidencePolicy
from v1.fall_state import Evidence, FallEvent, FallState, FallStateMachine
from v1.tracking import PersonTracker, TrackedPersonPose
from v1.video_source import FramePacket, SourceStatus
@dataclass(frozen=True)
class PersonAnalysis:
tracked_pose: TrackedPersonPose
pose_evidence: PoseEvidence
state: FallState
@dataclass(frozen=True)
class FrameAnalysis:
packet: FramePacket
people: Tuple[PersonAnalysis, ...]
events: Tuple[FallEvent, ...]
class FallPipeline:
"""Run one source frame through the V1 event decision flow."""
def __init__(
self,
pose_adapter,
tracker: PersonTracker,
policy: FallEvidencePolicy,
state_machine: FallStateMachine,
keypoint_confidence_threshold: float,
) -> None:
if not 0.0 <= keypoint_confidence_threshold <= 1.0:
raise ValueError("keypoint_confidence_threshold must be between 0 and 1")
self._pose_adapter = pose_adapter
self._tracker = tracker
self._policy = policy
self._state_machine = state_machine
self._keypoint_confidence_threshold = float(keypoint_confidence_threshold)
self._previous_evidence: Dict[str, PoseEvidence] = {}
self._active_track_ids = set()
@classmethod
def from_config(cls, config: AppConfig, pose_adapter) -> "FallPipeline":
"""Create one immutable runtime decision flow from validated config."""
return cls(
pose_adapter=pose_adapter,
tracker=PersonTracker(),
policy=FallEvidencePolicy(config.event.suspect_window_seconds),
state_machine=FallStateMachine(
confirm_window_seconds=config.event.confirm_window_seconds,
recovery_window_seconds=config.event.recovery_window_seconds,
cooldown_seconds=config.event.cooldown_seconds,
config_version=config.runtime_config_version,
),
keypoint_confidence_threshold=config.event.keypoint_confidence_threshold,
)
def process(self, packet: FramePacket) -> FrameAnalysis:
if packet.status is not SourceStatus.CONNECTED or packet.image is None:
events = self._reject_active_tracks(packet.timestamp_monotonic)
return FrameAnalysis(packet=packet, people=(), events=tuple(events))
height, width = packet.image.shape[:2]
poses = self._pose_adapter.infer(packet.image)
tracked_poses = self._tracker.update(
poses,
detected_at_monotonic=packet.timestamp_monotonic,
frame_size=(width, height),
)
current_ids = {tracked.track_id for tracked in tracked_poses}
events = self._reject_missing_tracks(current_ids, packet.timestamp_monotonic)
people = []
for tracked in tracked_poses:
quality = assess_pose_quality(
tracked.pose, threshold=self._keypoint_confidence_threshold
)
pose_evidence = extract_evidence(
tracked.pose,
quality=quality,
previous=self._previous_evidence.get(tracked.track_id),
)
state_before = self._state_machine.state_of(tracked.track_id)
evidence = self._policy.evaluate(
tracked.track_id,
pose_evidence,
now=packet.timestamp_monotonic,
state=state_before,
)
events.extend(
self._state_machine.update(
tracked.track_id, evidence, now=packet.timestamp_monotonic
)
)
if pose_evidence.accepted:
self._previous_evidence[tracked.track_id] = pose_evidence
else:
self._previous_evidence.pop(tracked.track_id, None)
people.append(
PersonAnalysis(
tracked_pose=tracked,
pose_evidence=pose_evidence,
state=self._state_machine.state_of(tracked.track_id),
)
)
self._active_track_ids = current_ids
return FrameAnalysis(packet=packet, people=tuple(people), events=tuple(events))
def _reject_missing_tracks(self, current_ids: set, now: float) -> list:
missing_ids = self._active_track_ids - current_ids
events = []
for track_id in sorted(missing_ids):
events.extend(self._reject_track(track_id, now))
return events
def _reject_active_tracks(self, now: float) -> list:
events = []
for track_id in sorted(self._active_track_ids):
events.extend(self._reject_track(track_id, now))
self._active_track_ids = set()
return events
def _reject_track(self, track_id: str, now: float) -> Sequence[FallEvent]:
rejected = Evidence(accepted=False, is_fall_candidate=False)
self._previous_evidence.pop(track_id, None)
self._policy.evaluate(
track_id,
PoseEvidence(False, False, False, None, None, None, "missing_pose"),
now=now,
state=self._state_machine.state_of(track_id),
)
return self._state_machine.update(track_id, rejected, now=now)