From 9f7d2be1acc5f2610338b1e97aac8fc08e852040 Mon Sep 17 00:00:00 2001 From: Martin Laasmaa Date: Thu, 11 Dec 2025 23:27:02 +0200 Subject: [PATCH] Updating the base model preset --- src/model/yolo_wrapper.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/src/model/yolo_wrapper.py b/src/model/yolo_wrapper.py index db95a2c..d5e31b3 100644 --- a/src/model/yolo_wrapper.py +++ b/src/model/yolo_wrapper.py @@ -80,7 +80,8 @@ class YOLOWrapper: Dictionary with training results """ if self.model is None: - self.load_model() + if not self.load_model(): + raise RuntimeError(f"Failed to load model from {self.model_path}") try: logger.info(f"Starting training: {name}") @@ -122,7 +123,8 @@ class YOLOWrapper: Dictionary with validation metrics """ if self.model is None: - self.load_model() + if not self.load_model(): + raise RuntimeError(f"Failed to load model from {self.model_path}") try: logger.info(f"Starting validation on {split} split") @@ -163,7 +165,8 @@ class YOLOWrapper: List of detection dictionaries """ if self.model is None: - self.load_model() + if not self.load_model(): + raise RuntimeError(f"Failed to load model from {self.model_path}") prepared_source, cleanup_path = self._prepare_source(source) @@ -211,7 +214,8 @@ class YOLOWrapper: Path to exported model """ if self.model is None: - self.load_model() + if not self.load_model(): + raise RuntimeError(f"Failed to load model from {self.model_path}") try: logger.info(f"Exporting model to {format} format")