mAP result: add IoU thresholds

roboflow · Aug 26, 2024 · b475c21 · b475c21
1 parent 2221d2b
commit b475c21
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/supervision/metrics/mean_average_precision.py b/supervision/metrics/mean_average_precision.py
@@ -234,6 +234,7 @@ def _compute(
             average_precisions = np.empty((0, len(iou_thresholds)), dtype=np.float32)
 
         return MeanAveragePrecisionResult(
+            iou_thresholds=iou_thresholds,
             map50_95=map50_95,
             map50=map50,
             map75=map75,
@@ -355,6 +356,7 @@ def _average_precisions_per_class(
 
 @dataclass
 class MeanAveragePrecisionResult:
+    iou_thresholds: np.ndarray
     map50_95: float
     map50: float
     map75: float