Allow agnostic NMS in validation for OBB, Pose, Segment and NAS (#16251)

2024-09-14 01:35:33 +08:00 · 2024-09-14 01:35:33 +08:00 · 6c3297623b
commit 6c3297623b
parent bf2b221d11
4 changed files with 4 additions and 4 deletions
--- 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,
        )