diff --git a/requirements-dev.txt b/requirements-dev.txt index 968d2e6..d877777 100644 --- 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 index 2056779..264c7fe 100644 --- a/requirements.txt +++ b/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 diff --git a/yolo/tools/solver.py b/yolo/tools/solver.py index a5efb6b..41e7aba 100644 --- 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 index 08973d0..28a5362 100644 --- 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")