From e8dcc3075480392661a6f06fc491844a587eb494 Mon Sep 17 00:00:00 2001 From: Clea Parcerisas Date: Mon, 21 Aug 2023 09:46:04 +0200 Subject: [PATCH] changed the weights for the fitness function --- ultralytics/utils/metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/utils/metrics.py b/ultralytics/utils/metrics.py index 7e393ae..6827453 100644 --- a/ultralytics/utils/metrics.py +++ b/ultralytics/utils/metrics.py @@ -625,7 +625,7 @@ class Metric(SimpleClass): def fitness(self): """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() def update(self, results):