Skip to content

NMS for DEKR models#2907

Merged
maximpavliv merged 5 commits intomainfrom
niels/dekr_nms
Mar 19, 2025
Merged

NMS for DEKR models#2907
maximpavliv merged 5 commits intomainfrom
niels/dekr_nms

Conversation

@n-poulsen
Copy link
Copy Markdown
Contributor

Implements pose non-maximum suppression for DEKR as done in the official DEKR implementation. Some slight changes were made to the code so it could be applied on batches of predictions.

@AlexEMG AlexEMG requested a review from maximpavliv February 28, 2025 09:14
@n-poulsen n-poulsen changed the base branch from pytorch_dlc to main March 18, 2025 09:16
Copy link
Copy Markdown
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 !

@maximpavliv maximpavliv merged commit 8e9e0cb into main Mar 19, 2025
1 check passed
@MMathisLab MMathisLab deleted the niels/dekr_nms branch April 14, 2025 11:48
deruyter92 added a commit to deruyter92/DeepLabCut-live that referenced this pull request Jan 21, 2026
This commit Updates the DEKR predictor to follow the DeepLabCut implementation in version 3.0.0rc7, see
DeepLabCut/DeepLabCut#2907
MMathisLab pushed a commit to DeepLabCut/DeepLabCut-live that referenced this pull request Jan 22, 2026
* DEKRPredictor: add non-maximum suppression (NMS)

This commit Updates the DEKR predictor to follow the DeepLabCut implementation in version 3.0.0rc7, see
DeepLabCut/DeepLabCut#2907

* DEKRPredictor: speed up with vectorized operations

This commit updates the DEKRPredictor to follow the DeepLabCut implementation in version 3.0.0rc13.  see DeepLabCut/DeepLabCut#3121

* PartAffinityFieldPredictor (PAF): Speed up cost computation

This commit updates the PAF predictor to follow the DeepLabCut implementation in version 3.0.0.rc13. See
DeepLabCut/DeepLabCut#3117

* HeatmapPredictor (single animal): speed up with vecorized operations

This commit updates the `HeatmapPredictor` in single_predictor.py to follow the implementation in DeepLabCut 3.0.0rc13. See DeepLabCut/DeepLabCut#3110
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