Detection training zero-mAP fix (#192)

2023-01-10 12:18:39 +01:00
parent a8a70390ab
commit 0a7d86cd26
3 changed files with 3 additions and 2 deletions
--- a/ultralytics/yolo/v8/detect/train.py
+++ b/ultralytics/yolo/v8/detect/train.py
@ -60,7 +60,7 @@ class DetectionTrainer(BaseTrainer):
    def get_model(self, cfg=None, weights=None, verbose=True):
        model = DetectionModel(cfg, ch=3, nc=self.data["nc"], verbose=verbose)
        if weights:
-            model.load(model)
+            model.load(weights)

        return model