From 7da74600added4b7b03bd3908179e8d131a1f6ac Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Tue, 12 Nov 2024 17:00:30 +0800 Subject: [PATCH] Include FPs for images with no labels in confusion matrix (#17481) Co-authored-by: Glenn Jocher --- ultralytics/models/yolo/detect/val.py | 4 ++-- ultralytics/models/yolo/pose/val.py | 4 ++-- ultralytics/models/yolo/segment/val.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/ultralytics/models/yolo/detect/val.py b/ultralytics/models/yolo/detect/val.py index 05db8cba..31f0fdc0 100644 --- a/ultralytics/models/yolo/detect/val.py +++ b/ultralytics/models/yolo/detect/val.py @@ -155,8 +155,8 @@ class DetectionValidator(BaseValidator): # Evaluate if nl: stat["tp"] = self._process_batch(predn, bbox, cls) - if self.args.plots: - self.confusion_matrix.process_batch(predn, bbox, cls) + if self.args.plots: + self.confusion_matrix.process_batch(predn, bbox, cls) for k in self.stats.keys(): self.stats[k].append(stat[k]) diff --git a/ultralytics/models/yolo/pose/val.py b/ultralytics/models/yolo/pose/val.py index bdf17328..12d812a7 100644 --- a/ultralytics/models/yolo/pose/val.py +++ b/ultralytics/models/yolo/pose/val.py @@ -138,8 +138,8 @@ class PoseValidator(DetectionValidator): if nl: stat["tp"] = self._process_batch(predn, bbox, cls) stat["tp_p"] = self._process_batch(predn, bbox, cls, pred_kpts, pbatch["kpts"]) - if self.args.plots: - self.confusion_matrix.process_batch(predn, bbox, cls) + if self.args.plots: + self.confusion_matrix.process_batch(predn, bbox, cls) for k in self.stats.keys(): self.stats[k].append(stat[k]) diff --git a/ultralytics/models/yolo/segment/val.py b/ultralytics/models/yolo/segment/val.py index 30b63f97..d8fe4aae 100644 --- a/ultralytics/models/yolo/segment/val.py +++ b/ultralytics/models/yolo/segment/val.py @@ -135,8 +135,8 @@ class SegmentationValidator(DetectionValidator): stat["tp_m"] = self._process_batch( predn, bbox, cls, pred_masks, gt_masks, self.args.overlap_mask, masks=True ) - if self.args.plots: - self.confusion_matrix.process_batch(predn, bbox, cls) + if self.args.plots: + self.confusion_matrix.process_batch(predn, bbox, cls) for k in self.stats.keys(): self.stats[k].append(stat[k])