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

action-recog
Mohammed Yasin 2 months ago committed by fcakyon
parent ae22248abc
commit 9353e7d371
  1. 2
      ultralytics/models/nas/val.py
  2. 2
      ultralytics/models/yolo/obb/val.py
  3. 2
      ultralytics/models/yolo/pose/val.py
  4. 2
      ultralytics/models/yolo/segment/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,
)

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

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

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

Loading…
Cancel
Save