How can i get keypoints and keypoint_scores by the heatmap?

[hongin13]

Hi, i want to compare the difference between the keypoints on the existing model and the keypoints created directly by the heatmap, does anyone know where and what calculations are done to get keypoints and keypoint_scores?

[Ben-Louis]

Hi @hongin13, you can refer to the get_heatmap_maximum function at

mmpose/mmpose/codecs/utils/post_processing.py

Line 145 in 6d10b2e

def get_heatmap_maximum(heatmaps: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:

[hongin13]

I tried it and it came out well. Thank you for letting me know :)