diff --git a/src/otx/core/metrics/average_precision_3d.py b/src/otx/core/metrics/average_precision_3d.py index 95a3da397f4..155d9f47fc8 100644 --- a/src/otx/core/metrics/average_precision_3d.py +++ b/src/otx/core/metrics/average_precision_3d.py @@ -45,6 +45,7 @@ def reset(self) -> None: super().reset() self.preds: list[dict[str, np.array]] = [] self.targets: list[dict[str, np.array]] = [] + self.mean_ap.reset() def update(self, preds: list[dict[str, Tensor]], target: list[dict[str, Tensor]]) -> None: """Update total predictions and targets from given batch predicitons and targets.""" diff --git a/tests/integration/cli/test_export_inference.py b/tests/integration/cli/test_export_inference.py index 2e556210165..1d455616c4f 100644 --- a/tests/integration/cli/test_export_inference.py +++ b/tests/integration/cli/test_export_inference.py @@ -49,7 +49,7 @@ def fxt_local_seed() -> int: "zero_shot_visual_prompting": "test/f1-score", "action_classification": "test/accuracy", "keypoint_detection": "test/PCK", - "object_detection_3d": "test/mAP_bbox_3d", + "object_detection_3d": "test/AP_3d@0.5", }