Simplify argument names (#141)

2023-01-03 19:47:56 +01:00 · 2023-01-03 19:47:56 +01:00 · 3c18aa5278
commit 3c18aa5278
parent d74de2582c
9 changed files with 22 additions and 24 deletions
--- a/ultralytics/yolo/v8/detect/predict.py
+++ b/ultralytics/yolo/v8/detect/predict.py
@ -20,8 +20,8 @@ class DetectionPredictor(BasePredictor):

    def postprocess(self, preds, img, orig_img):
        preds = ops.non_max_suppression(preds,
-                                        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)