Skip to content

Commit fb1f722

Browse files
authored
fixing confidence computation (#6)
1 parent fc3b470 commit fb1f722

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

evaluation/eval.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,7 @@ def _preprocess_distance_and_confidence(pred_all, gt_all):
120120
pred_loc = pred_cls[:, :2].reshape([-1, 1, 2])
121121
gt_loc = gt_cls[:, :2].reshape([1, -1, 2])
122122
distance = np.linalg.norm(pred_loc - gt_loc, axis=2)
123-
confidence = pred_cls[:, 2]
123+
confidence = pred_cls[:, 3]
124124

125125
one_sample_result[cls_idx] = (distance, confidence)
126126

0 commit comments

Comments
 (0)