Skip to content

Commit

Permalink
fix inferencer
Browse files Browse the repository at this point in the history
  • Loading branch information
LareinaM committed Sep 19, 2023
1 parent ce8da8f commit 5bfbb27
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
3 changes: 1 addition & 2 deletions mmpose/apis/inferencers/mmpose_inferencer.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,8 +55,7 @@ class MMPoseInferencer(BaseMMPoseInferencer):
"""

preprocess_kwargs: set = {
'bbox_thr', 'nms_thr', 'bboxes', 'use_oks_tracking', 'tracking_thr',
'disable_norm_pose_2d'
'bbox_thr', 'nms_thr', 'bboxes', 'use_oks_tracking', 'tracking_thr'
}
forward_kwargs: set = {'disable_rebase_keypoint'}
visualize_kwargs: set = {
Expand Down
3 changes: 1 addition & 2 deletions mmpose/apis/inferencers/pose3d_inferencer.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,8 +68,7 @@ class Pose3DInferencer(BaseMMPoseInferencer):
"""

preprocess_kwargs: set = {
'bbox_thr', 'nms_thr', 'bboxes', 'use_oks_tracking', 'tracking_thr',
'disable_norm_pose_2d'
'bbox_thr', 'nms_thr', 'bboxes', 'use_oks_tracking', 'tracking_thr'
}
forward_kwargs: set = {'disable_rebase_keypoint'}
visualize_kwargs: set = {
Expand Down

0 comments on commit 5bfbb27

Please sign in to comment.