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

Apply default ConfusionMatrix conf=0.25 (#4813)

Browse Source
This commit is contained in:
Glenn Jocher 2023-09-10 20:50:03 +02:00 committed by GitHub
parent dbc53f6741
commit dd0782bd8d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/utils/metrics.py
View File

@ -189,7 +189,7 @@ class ConfusionMatrix:
self.task = task self.task = task
self.matrix = np.zeros((nc + 1, nc + 1)) if self.task == 'detect' else np.zeros((nc, nc)) self.matrix = np.zeros((nc + 1, nc + 1)) if self.task == 'detect' else np.zeros((nc, nc))
self.nc = nc # number of classes self.nc = nc # number of classes
self.conf = 0.25 if conf is None else conf # argument may be None from default cfg self.conf = 0.25 if conf in (None, 0.001) else conf # apply 0.25 if default val conf is passed
self.iou_thres = iou_thres self.iou_thres = iou_thres
def process_cls_preds(self, preds, targets): def process_cls_preds(self, preds, targets):