diff --git a/faster_coco_eval/core/faster_eval_api.py b/faster_coco_eval/core/faster_eval_api.py index 3d3e3a4..2c56290 100644 --- a/faster_coco_eval/core/faster_eval_api.py +++ b/faster_coco_eval/core/faster_eval_api.py @@ -235,7 +235,7 @@ def extended_metrics(self): iou_thrs, rec_thrs = self.params.iouThrs, self.params.recThrs # Indices for IoU=0.50, first area, and last max dets - iou50_idx, area_idx, maxdet_idx = (int(np.argwhere(np.isclose(iou_thrs, 0.50))), 0, -1) + iou50_idx, area_idx, maxdet_idx = (np.argwhere(np.isclose(iou_thrs, 0.50)).item(), 0, -1) P = self.eval["precision"] S = self.eval["scores"]