search

yangsenius / TransPose

PyTorch Implementation for "TransPose: Keypoint localization via Transformer", ICCV 2021.
https://github.com/yangsenius/TransPose/releases/download/paper/transpose.pdf
MIT License
353 stars 56 forks source link

Get the confidence Score of each keypoint #43

Open adnantariq18 opened 2 years ago

adnantariq18 commented 2 years ago

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 commented 2 years ago

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

adnantariq18 commented 2 years ago

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