Skip to content

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.

3 participants