We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fc3b470 commit fb1f722Copy full SHA for fb1f722
evaluation/eval.py
@@ -120,7 +120,7 @@ def _preprocess_distance_and_confidence(pred_all, gt_all):
120
pred_loc = pred_cls[:, :2].reshape([-1, 1, 2])
121
gt_loc = gt_cls[:, :2].reshape([1, -1, 2])
122
distance = np.linalg.norm(pred_loc - gt_loc, axis=2)
123
- confidence = pred_cls[:, 2]
+ confidence = pred_cls[:, 3]
124
125
one_sample_result[cls_idx] = (distance, confidence)
126
0 commit comments