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/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,
        )