Revert MAP implementation to pycocotools backend

CHANGELOG.md

@@ -155,6 +155,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Changed calculation in `PearsonCorrCoeff` to be more robust in certain cases ([#1729](https://github.com/Lightning-AI/torchmetrics/pull/1729))
 
+
+- Changed `MeanAveragePrecision` to `pycocotools` backend ([#1832](https://github.com/Lightning-AI/torchmetrics/pull/1832))
+
+
 ### Deprecated
 
 - Deprecated domain metrics import from package root (

pyproject.toml

@@ -184,6 +184,7 @@ module = [
  "torchmetrics.classification.roc",
  "torchmetrics.classification.specificity",
  "torchmetrics.classification.stat_scores",
+ "torchmetrics.detection._mean_ap",
  "torchmetrics.detection.mean_ap",
  "torchmetrics.functional.classification.calibration_error",
  "torchmetrics.functional.classification.confusion_matrix",

tests/unittests/conftest.py

@@ -17,10 +17,11 @@
 
 import pytest
 import torch
-from torch.multiprocessing import Pool, set_start_method
+from torch.multiprocessing import Pool, set_sharing_strategy, set_start_method
 
 with contextlib.suppress(RuntimeError):
  set_start_method("spawn")
+ set_sharing_strategy("file_system")
 
 NUM_PROCESSES = 2 # torch.cuda.device_count() if torch.cuda.is_available() else 2
 NUM_BATCHES = 2 * NUM_PROCESSES # Need to be divisible with the number of processes

tests/unittests/detection/__init__.py

@@ -3,3 +3,6 @@
 from unittests import _PATH_ROOT
 
 _SAMPLE_DETECTION_SEGMENTATION = os.path.join(_PATH_ROOT, "_data", "detection", "instance_segmentation_inputs.json")
+_DETECTION_VAL = os.path.join(_PATH_ROOT, "_data", "detection", "instances_val2014_100.json")
+_DETECTION_BBOX = os.path.join(_PATH_ROOT, "_data", "detection", "instances_val2014_fakebbox100_results.json")
+_DETECTION_SEGM = os.path.join(_PATH_ROOT, "_data", "detection", "instances_val2014_fakesegm100_results.json")