diff --git a/mmpose/core/post_processing/group.py b/mmpose/core/post_processing/group.py index 11614c6bea..6235dbc111 100644 --- a/mmpose/core/post_processing/group.py +++ b/mmpose/core/post_processing/group.py @@ -376,7 +376,7 @@ def parse(self, heatmaps, tags, adjust=True, refine=True): tuple: A tuple containing keypoint grouping results. - results (list(np.ndarray)): Pose results. - - scores (list): Score of people. + - scores (list/list(np.ndarray)): Score of people. """ results = self.match(**self.top_k(heatmaps, tags)) diff --git a/mmpose/core/post_processing/nms.py b/mmpose/core/post_processing/nms.py index 37781c2b28..86a0ab35e0 100644 --- a/mmpose/core/post_processing/nms.py +++ b/mmpose/core/post_processing/nms.py @@ -94,6 +94,7 @@ def oks_nms(kpts_db, thr, sigmas=None, vis_thr=None, score_per_joint=False): thr: Retain overlap < thr. sigmas: standard deviation of keypoint labelling. vis_thr: threshold of the keypoint visibility. + score_per_joint: the input scores (in kpts_db) are per joint scores Returns: np.ndarray: indexes to keep. @@ -164,6 +165,7 @@ def soft_oks_nms(kpts_db, thr: retain oks overlap < thr. max_dets: max number of detections to keep. sigmas: Keypoint labelling uncertainty. + score_per_joint: the input scores (in kpts_db) are per joint scores Returns: np.ndarray: indexes to keep.