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

changed the weights for the fitness function

Browse Source
This commit is contained in:
Clea Parcerisas
2023-08-21 09:46:04 +02:00
parent 3acead7e79
commit e8dcc30754
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ultralytics/utils/metrics.py
View File

@ -625,7 +625,7 @@ class Metric(SimpleClass):
def fitness(self): def fitness(self):
"""Model fitness as a weighted combination of metrics.""" """Model fitness as a weighted combination of metrics."""
w = [0.0, 0.0, 0.1, 0.9] # weights for [P, R, mAP@0.5, mAP@0.5:0.95] w = [0.1, 0.4, 0.5, 0.0] # weights for [P, R, mAP@0.5, mAP@0.5:0.95]
return (np.array(self.mean_results()) * w).sum() return (np.array(self.mean_results()) * w).sum()
def update(self, results): def update(self, results):