Debug: set default dataset path

yihong1120 · Jun 29, 2024 · 23dac0b · 23dac0b
1 parent c6cc556
commit 23dac0b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/YOLOv8-Evaluation/evaluate_yolov8.py b/examples/YOLOv8-Evaluation/evaluate_yolov8.py
@@ -27,8 +27,9 @@ def evaluate(self) -> Dict[str, Any]:
         Returns:
             Dict[str, Any]: The results from the model evaluation.
         """
+        print(f"Evaluating model with data path: {self.data_path}")
         # The 'val' method is used for evaluation, 'test' could also be used as per the requirement.
-        return self.model.val()
+        return self.model.val(data=self.data_path)
 
 def parse_arguments() -> argparse.Namespace:
     """