Simplify argument names (#141)

2023-01-03 19:47:56 +01:00
parent d74de2582c
commit 3c18aa5278
9 changed files with 22 additions and 24 deletions
--- a/ultralytics/yolo/v8/segment/predict.py
+++ b/ultralytics/yolo/v8/segment/predict.py
@ -14,8 +14,8 @@ class SegmentationPredictor(DetectionPredictor):
        masks = []
        # TODO: filter by classes
        p = ops.non_max_suppression(preds[0],
-                                    self.args.conf_thres,
-                                    self.args.iou_thres,
+                                    self.args.conf,
+                                    self.args.iou,
                                    agnostic=self.args.agnostic_nms,
                                    max_det=self.args.max_det,
                                    nm=32)
--- a/ultralytics/yolo/v8/segment/val.py
+++ b/ultralytics/yolo/v8/segment/val.py
@ -60,8 +60,8 @@ class SegmentationValidator(DetectionValidator):

    def postprocess(self, preds):
        p = ops.non_max_suppression(preds[0],
-                                    self.args.conf_thres,
-                                    self.args.iou_thres,
+                                    self.args.conf,
+                                    self.args.iou,
                                    labels=self.lb,
                                    multi_label=True,
                                    agnostic=self.args.single_cls,