diff --git a/ultralytics/models/nas/val.py b/ultralytics/models/nas/val.py index 7de446d4..c4fb73b6 100644 --- a/ultralytics/models/nas/val.py +++ b/ultralytics/models/nas/val.py @@ -44,7 +44,7 @@ class NASValidator(DetectionValidator): self.args.iou, labels=self.lb, multi_label=False, - agnostic=self.args.single_cls, + agnostic=self.args.single_cls or self.args.agnostic_nms, max_det=self.args.max_det, max_time_img=0.5, ) diff --git a/ultralytics/models/yolo/obb/val.py b/ultralytics/models/yolo/obb/val.py index b4048af4..93bb2bfa 100644 --- a/ultralytics/models/yolo/obb/val.py +++ b/ultralytics/models/yolo/obb/val.py @@ -45,7 +45,7 @@ class OBBValidator(DetectionValidator): labels=self.lb, nc=self.nc, multi_label=True, - agnostic=self.args.single_cls, + agnostic=self.args.single_cls or self.args.agnostic_nms, max_det=self.args.max_det, rotated=True, ) diff --git a/ultralytics/models/yolo/pose/val.py b/ultralytics/models/yolo/pose/val.py index cba42a6e..bdf17328 100644 --- a/ultralytics/models/yolo/pose/val.py +++ b/ultralytics/models/yolo/pose/val.py @@ -69,7 +69,7 @@ class PoseValidator(DetectionValidator): self.args.iou, labels=self.lb, multi_label=True, - agnostic=self.args.single_cls, + agnostic=self.args.single_cls or self.args.agnostic_nms, max_det=self.args.max_det, nc=self.nc, ) diff --git a/ultralytics/models/yolo/segment/val.py b/ultralytics/models/yolo/segment/val.py index 36ccd9c2..30b63f97 100644 --- a/ultralytics/models/yolo/segment/val.py +++ b/ultralytics/models/yolo/segment/val.py @@ -76,7 +76,7 @@ class SegmentationValidator(DetectionValidator): self.args.iou, labels=self.lb, multi_label=True, - agnostic=self.args.single_cls, + agnostic=self.args.single_cls or self.args.agnostic_nms, max_det=self.args.max_det, nc=self.nc, )