Skip to content

Improve tracklet creation for ID-only tracking#2903

Merged
n-poulsen merged 6 commits intopytorch_dlcfrom
niels/id_tracking
Feb 26, 2025
Merged

Improve tracklet creation for ID-only tracking#2903
n-poulsen merged 6 commits intopytorch_dlcfrom
niels/id_tracking

Conversation

@n-poulsen
Copy link
Contributor

This pull request fixes tracklet creation with identity_only=True. It

  • only keeps valid predictions to assign identity to
  • removes poses with confidence below the inference pcutoff
  • updates the way the softmax is computed to assign each assembly (total score across keypoints with a given identity instead of average score)

It also fixes identity metric computation when NaNs are in the predictions.

Copy link
Contributor

@maximpavliv maximpavliv left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks @n-poulsen !

@n-poulsen n-poulsen merged commit d149913 into pytorch_dlc Feb 26, 2025
1 check passed
@n-poulsen n-poulsen deleted the niels/id_tracking branch February 26, 2025 13:58
@n-poulsen n-poulsen mentioned this pull request Feb 26, 2025
2 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

Comments