fix udp_combined decode bug (open-mmlab#1565)

evendrow · Dec 22, 2022 · 44f5eb9 · 44f5eb9
1 parent dbfa68b
commit 44f5eb9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmpose/core/evaluation/top_down_eval.py b/mmpose/core/evaluation/top_down_eval.py
@@ -579,7 +579,7 @@ def keypoints_from_heatmaps(heatmaps,
             offset_y = heatmaps[:, 2::3, :].flatten() * valid_radius
             heatmaps = heatmaps[:, ::3, :]
             preds, maxvals = _get_max_preds(heatmaps)
-            index = preds[..., 0] + preds[..., 1] * W
+            index = (preds[..., 0] + preds[..., 1] * W).flatten()
             index += W * H * np.arange(0, N * K / 3)
             index = index.astype(int).reshape(N, K // 3, 1)
             preds += np.concatenate((offset_x[index], offset_y[index]), axis=2)