diff --git a/supervision/metrics/mean_average_precision.py b/supervision/metrics/mean_average_precision.py index 2429c1898..58d4764c1 100644 --- a/supervision/metrics/mean_average_precision.py +++ b/supervision/metrics/mean_average_precision.py @@ -9,7 +9,11 @@ from supervision.config import ORIENTED_BOX_COORDINATES from supervision.detection.core import Detections -from supervision.detection.utils import box_iou_batch, mask_iou_batch, oriented_box_iou_batch +from supervision.detection.utils import ( + box_iou_batch, + mask_iou_batch, + oriented_box_iou_batch, +) from supervision.draw.color import LEGACY_COLOR_PALETTE from supervision.metrics.core import Metric, MetricTarget from supervision.metrics.utils.object_size import ( @@ -177,9 +181,13 @@ def _compute( elif self._metric_target == MetricTarget.MASKS: iou = mask_iou_batch(target_contents, prediction_contents) elif self._metric_target == MetricTarget.ORIENTED_BOUNDING_BOXES: - iou = oriented_box_iou_batch(target_contents, prediction_contents) + iou = oriented_box_iou_batch( + target_contents, prediction_contents + ) else: - raise ValueError(f"Unsupported metric target: {self._metric_target}") + raise ValueError( + f"Unsupported metric target: {self._metric_target}" + ) matches = self._match_detection_batch( predictions.class_id, targets.class_id, iou, iou_thresholds