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

requirements-dev.txt

@@ -3,5 +3,4 @@ gradio
 pytest
 pytest-cov
 pre-commit
-pycocotools
 tensorboard

requirements.txt

@@ -1,4 +1,5 @@
 einops
+faster-coco-eval
 graphviz
 hydra-core
 lightning
@@ -6,7 +7,6 @@ loguru
 numpy
 opencv-python
 Pillow
-pycocotools
 requests
 rich
 torch

yolo/tools/solver.py

@@ -30,7 +30,7 @@ class ValidateModel(BaseModel):
             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

yolo/utils/logging_utils.py

@@ -237,6 +237,9 @@ def setup_logger(logger_name, quite=False):
         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")