changed the weights for the fitness function

2 years ago · e8dcc30754
parent 3acead7e79
commit e8dcc30754
1 changed files with 1 additions and 1 deletions
--- 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):