Allow agnostic_nms option for validation (#14675)

2024-07-25 19:34:15 +08:00 · 2024-07-25 19:34:15 +08:00 · 15266d37a1
commit 15266d37a1
parent 9df18ca9ef
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/models/yolo/detect/val.py
+++ b/ultralytics/models/yolo/detect/val.py
@ -97,7 +97,7 @@ class DetectionValidator(BaseValidator):
            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,
        )