xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify argument names (#141)

Browse Source
This commit is contained in:
Glenn Jocher
2023-01-03 19:47:56 +01:00
committed by GitHub
parent d74de2582c
commit 3c18aa5278
9 changed files with 22 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ultralytics/yolo/v8/detect/predict.py
View File

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

4
ultralytics/yolo/v8/detect/val.py
View File

@ -58,8 +58,8 @@ class DetectionValidator(BaseValidator):
def postprocess(self, preds):
preds = ops.non_max_suppression(preds,
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,

4
ultralytics/yolo/v8/segment/predict.py
View File

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

4
ultralytics/yolo/v8/segment/val.py
View File

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