Skip to content

Commit f3d3118

Browse files
committed
Use runner pose directly in DLC processor
Switch pose handling to validate `self._dlc.pose` from `runner.get_pose` instead of calling the private `_post_process_pose` path. The emitted `PosePacket.raw` now carries the validated pose array, keeping packet contents aligned with the processed keypoints.
1 parent 275aedc commit f3d3118

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

‎dlclivegui/services/dlc_processor.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -590,19 +590,19 @@ def _process_frame(
590590
with self._timing.measure("DLC.runner.get_pose"):
591591
self._dlc.pose = self._dlc.runner.get_pose(processed_frame)
592592

593-
with self._timing.measure("DLC.post_process_pose"):
594-
raw_pose: Any = self._dlc._post_process_pose(processed_frame, frame_time=timestamp)
593+
# with self._timing.measure("DLC.post_process_pose"):
594+
# raw_pose: Any = self._dlc._post_process_pose(processed_frame, frame_time=timestamp)
595595

596596
inference_time = time.perf_counter() - inference_start
597597
with self._timing.measure("DLC.validate_pose"):
598-
pose_arr: np.ndarray = validate_pose_array(raw_pose, source_backend=PoseBackends.DLC_LIVE)
598+
pose_arr: np.ndarray = validate_pose_array(self._dlc.pose, source_backend=PoseBackends.DLC_LIVE)
599599
pose_packet = PosePacket(
600600
schema_version=0,
601601
keypoints=pose_arr,
602602
keypoint_names=None,
603603
individual_ids=None,
604604
source=PoseSource(backend=PoseBackends.DLC_LIVE, model_type=self._settings.model_type),
605-
raw=raw_pose,
605+
raw=pose_arr,
606606
)
607607

608608
processor_overhead = 0.0

0 commit comments

Comments
 (0)