♻️ [Update] pycoco with faster coco, fast validate

WongKinYiu · Dec 23, 2024 · 3e180a7 · 3e180a7
1 parent b96c8ea
commit 3e180a7
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/requirements-dev.txt b/requirements-dev.txt
@@ -3,5 +3,4 @@ gradio
 pytest
 pytest-cov
 pre-commit
-pycocotools
 tensorboard
diff --git a/requirements.txt b/requirements.txt
@@ -1,12 +1,12 @@
 einops
+faster-coco-eval
 graphviz
 hydra-core
 lightning
 loguru
 numpy
 opencv-python
 Pillow
-pycocotools
 requests
 rich
 torch

diff --git a/yolo/tools/solver.py b/yolo/tools/solver.py
@@ -30,7 +30,7 @@ def __init__(self, cfg: Config):
             self.validation_cfg = self.cfg.task
         else:
             self.validation_cfg = self.cfg.task.validation
-        self.metric = MeanAveragePrecision(iou_type="bbox", box_format="xyxy")
+        self.metric = MeanAveragePrecision(iou_type="bbox", box_format="xyxy", backend="faster_coco_eval")
         self.metric.warn_on_many_detections = False
         self.val_loader = create_dataloader(self.validation_cfg.data, self.cfg.dataset, self.validation_cfg.task)
         self.ema = self.model

diff --git a/yolo/utils/logging_utils.py b/yolo/utils/logging_utils.py
@@ -237,6 +237,9 @@ def format(self, record, emoji=":high_voltage:"):
         if quite:
             rich_logger.setLevel(logging.ERROR)
 
+    coco_logger = logging.getLogger("faster_coco_eval.core.cocoeval")
+    coco_logger.setLevel(logging.ERROR)
+
 
 def setup(cfg: Config):
     quite = hasattr(cfg, "quite")