Skip to content

Commit

Permalink
fix and op bug (open-mmlab#2286)
Browse files Browse the repository at this point in the history
  • Loading branch information
FishBigOcean authored and Tau-J committed Apr 25, 2023
1 parent 809b84a commit 49b4da1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion mmpose/evaluation/functional/nms.py
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ def oks_iou(g: np.ndarray,
dy = yd - yg
e = (dx**2 + dy**2) / vars / ((a_g + a_d[n_d]) / 2 + np.spacing(1)) / 2
if vis_thr is not None:
ind = list(vg > vis_thr) and list(vd > vis_thr)
ind = list((vg > vis_thr) & (vd > vis_thr))
e = e[ind]
ious[n_d] = np.sum(np.exp(-e)) / len(e) if len(e) != 0 else 0.0
return ious
Expand Down

0 comments on commit 49b4da1

Please sign in to comment.