How to Obtain Confidence Scores for Key Points from GTPT?

[wenkaier]

Hi,

I'm working with the GTPT model and I've managed to identify key points using its capabilities. However, I'm struggling to understand how to extract or estimate the confidence scores associated with these key points. Could you please provide guidance on:

• The method or function to call within the GTPT library/API to get these scores.
• Any specific parameters or settings that need to be configured to enable confidence score calculation.
• Examples or documentation that can help me better understand the process.

Thank you in advance for your assistance!

Best regards

[haonanwang0522]

Thank you for your enthusiastic interest in our work. You can locate the confidence scores in the file lib/core/function.py, specifically on lines 188 to 193.

max_val_x, preds_x = output_x.max(2,keepdim=True)
max_val_y, preds_y = output_y.max(2,keepdim=True)

mask = max_val_x > max_val_y
max_val_x[mask] = max_val_y[mask]
maxvals = max_val_x.cpu().numpy()