[Fix] docstring

piercus · Feb 5, 2022 · a667627 · a667627
1 parent ce3a0e9
commit a667627
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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.