Skip to content

Commit

Permalink
fix typo
Browse files Browse the repository at this point in the history
  • Loading branch information
Tau-J committed Oct 8, 2023
1 parent 5be4146 commit 3dac3de
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions mmpose/evaluation/metrics/coco_metric.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ def __init__(self,
nms_mode: str = 'oks_nms',
nms_thr: float = 0.9,
format_only: bool = False,
pred_conveter: Dict = None,
pred_converter: Dict = None,
gt_converter: Dict = None,
outfile_prefix: Optional[str] = None,
collect_device: str = 'cpu',
Expand Down Expand Up @@ -148,7 +148,7 @@ def __init__(self,

self.format_only = format_only
self.outfile_prefix = outfile_prefix
self.pred_conveter = pred_conveter
self.pred_converter = pred_converter
self.gt_converter = gt_converter

@property
Expand Down Expand Up @@ -403,9 +403,10 @@ def compute_metrics(self, results: list) -> Dict[str, float]:
for pred in preds:
img_id = pred['img_id']

if self.pred_conveter is not None:
pred = transform_ann(pred, self.pred_conveter['num_keypoints'],
self.pred_conveter['mapping'])
if self.pred_converter is not None:
pred = transform_ann(pred,
self.pred_converter['num_keypoints'],
self.pred_converter['mapping'])

for idx, keypoints in enumerate(pred['keypoints']):

Expand Down

0 comments on commit 3dac3de

Please sign in to comment.