xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Default confidences predict 0.25, val 0.001 (#149)

Browse Source
This commit is contained in:
Glenn Jocher 2023-01-05 15:56:52 +01:00 committed by GitHub
parent 3cbf3ec455
commit 314da263c7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/yolo/configs/default.yaml
View File

@ -39,7 +39,7 @@ dropout: False # use dropout regularization
val: True # validate/test during training val: True # validate/test during training
save_json: False # save results to JSON file save_json: False # save results to JSON file
save_hybrid: False # save hybrid version of labels (labels + additional predictions) save_hybrid: False # save hybrid version of labels (labels + additional predictions)
conf: 0.001 # object confidence threshold for detection conf: null # object confidence threshold for detection (default 0.25 predict, 0.001 val)
iou: 0.7 # intersection over union (IoU) threshold for NMS iou: 0.7 # intersection over union (IoU) threshold for NMS
max_det: 300 # maximum number of detections per image max_det: 300 # maximum number of detections per image
half: False # use half precision (FP16) half: False # use half precision (FP16)

3
ultralytics/yolo/engine/predictor.py
View File

@ -77,7 +77,8 @@ class BasePredictor:
name = self.args.name or f"{self.args.mode}" name = self.args.name or f"{self.args.mode}"
self.save_dir = increment_path(Path(project) / name, exist_ok=self.args.exist_ok) self.save_dir = increment_path(Path(project) / name, exist_ok=self.args.exist_ok)
(self.save_dir / 'labels' if self.args.save_txt else self.save_dir).mkdir(parents=True, exist_ok=True) (self.save_dir / 'labels' if self.args.save_txt else self.save_dir).mkdir(parents=True, exist_ok=True)
if self.args.conf is None:
self.args.conf = 0.25 # default conf=0.25
self.done_setup = False self.done_setup = False
# Usable if setup is done # Usable if setup is done

3
ultralytics/yolo/engine/validator.py
View File

@ -65,6 +65,9 @@ class BaseValidator:
exist_ok=self.args.exist_ok if RANK in {-1, 0} else True) exist_ok=self.args.exist_ok if RANK in {-1, 0} else True)
(self.save_dir / 'labels' if self.args.save_txt else self.save_dir).mkdir(parents=True, exist_ok=True) (self.save_dir / 'labels' if self.args.save_txt else self.save_dir).mkdir(parents=True, exist_ok=True)
if self.args.conf is None:
self.args.conf = 0.001 # default conf=0.001
self.callbacks = defaultdict(list, {k: [v] for k, v in callbacks.default_callbacks.items()}) # add callbacks self.callbacks = defaultdict(list, {k: [v] for k, v in callbacks.default_callbacks.items()}) # add callbacks
@smart_inference_mode() @smart_inference_mode()