Skip to content

ExtendedKeypointMetrics and choose between depthai and luxonis-train post-processing in the keypoint parser#4

Draft
dtronmans wants to merge 2 commits intomainfrom
feat/extend-metrics
Draft

ExtendedKeypointMetrics and choose between depthai and luxonis-train post-processing in the keypoint parser#4
dtronmans wants to merge 2 commits intomainfrom
feat/extend-metrics

Conversation

@dtronmans
Copy link
Copy Markdown

@dtronmans dtronmans commented Apr 22, 2026

Purpose

  • Can pass sigmas and area_factor into ExtendedKeypointMetrics for the metrics calculation to match luxonis-train config
  • The keypoint parser now accepts postprocessing: luxonis_train (default is depthai, which keeps the current behavior). depthai: normalized keypoints, luxonis_train: pixel-space
  • Arbitrary number of keypoints per class and arbitrary number of classes

Change with output names (this is an optional/separate change): just match the ONNX outputs by name (like kpt_output* like in depthai-nodes) instead of relying on positional ordering

Example parser in the config:

parser:
  name: YOLOKeypointDetectionParser
  params:
    subtype: yolov8
    n_classes: 3
    conf_thres: 0.03
    iou_thres: 0.65
    max_det: 300
    postprocessing: luxonis_train

Example keypoint metric:

metrics:
  metrics:
    - name: ExtendedKeypointMetrics
      params:
        kpt_oks_sigmas: [0.08, 0.08, 0.08, 0.08]
        area_factor: 1.0

Specification

Dependencies & Potential Impact

Deployment Plan

Testing & Validation

@dtronmans dtronmans marked this pull request as draft April 22, 2026 17:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant