fixing confidence computation (#6)

lunit-io · Jul 29, 2023 · fb1f722 · fb1f722
1 parent fc3b470
commit fb1f722
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/evaluation/eval.py b/evaluation/eval.py
@@ -120,7 +120,7 @@ def _preprocess_distance_and_confidence(pred_all, gt_all):
                 pred_loc = pred_cls[:, :2].reshape([-1, 1, 2])
                 gt_loc = gt_cls[:, :2].reshape([1, -1, 2])
                 distance = np.linalg.norm(pred_loc - gt_loc, axis=2)
-                confidence = pred_cls[:, 2]
+                confidence = pred_cls[:, 3]
 
             one_sample_result[cls_idx] = (distance, confidence)