fix and op bug (open-mmlab#2286)

Tau-J · Apr 25, 2023 · d386aed · d386aed
1 parent a464ac8
commit d386aed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmpose/evaluation/functional/nms.py b/mmpose/evaluation/functional/nms.py
@@ -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