Skip to content

Commit

Permalink
refine comment
Browse files Browse the repository at this point in the history
  • Loading branch information
Ben-Louis committed Dec 11, 2023
1 parent e669d16 commit c3ec981
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mmdeploy/codebase/mmpose/models/heads/rtmo_head.py
Original file line number Diff line number Diff line change
Expand Up @@ -92,7 +92,7 @@ def predict(self,
kpt_vis = flatten_kpt_vis[batch_inds, nms_indices, ...]
grids = self.flatten_priors[nms_indices, ...]

# filter and decode keypoints
# decode keypoints
bbox_cs = torch.cat(bbox_xyxy2cs(dets[..., :4], self.bbox_padding), dim=-1)
keypoints = self.dcc.forward_test(pose_vecs, bbox_cs, grids)
pred_kpts = torch.cat([keypoints, kpt_vis.unsqueeze(-1)], dim=-1)
Expand Down

0 comments on commit c3ec981

Please sign in to comment.