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