Skip to content

Commit

Permalink
Fixed indexing bug recall_per_iou
Browse files Browse the repository at this point in the history
Fixed an indexing bug when calculating the recall_per_iou
  • Loading branch information
ChristofferEdlund authored Mar 21, 2022
1 parent 11359dc commit 1915b25
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion code/coco_evaluation_resnest.py
Original file line number Diff line number Diff line change
Expand Up @@ -654,7 +654,7 @@ def _summarizeKps():

# IoU threshold | instances | Categories | areas | max dets
pre_per_iou = [precisions[iou_idx, :, :, 0, -1].mean() for iou_idx in range(precisions.shape[0])]
rec_pre_iou = [recalls[iou_idx, :, :, 0, -1].mean() for iou_idx in range(recalls.shape[0])]
rec_pre_iou = [recalls[iou_idx, :, 0, -1].mean() for iou_idx in range(recalls.shape[0])]

print(f"Precision and Recall per iou: {coco_eval.params.iouThrs}")
print(np.round(np.array(pre_per_iou), 4))
Expand Down

0 comments on commit 1915b25

Please sign in to comment.