Get the confidence Score of each keypoint

[adnantariq18]

hi there. Is it possible to get the confidence score for the predicted key point for pre-trained model and custom trained model
what modifications are required to get the probability or confidence scores with the predicted keypoint

[yangsenius]

You can take the heatmap values of the maximum activation positions as the confidences for the keypoints.

[adnantariq18]

i think i didn't understand correctly can you elaborate plz .. are you talking about the function "get_max_pred function" ? If so what modification do i need