From 23dac0bd4d9411ce849ad3de292916510d9c7bda Mon Sep 17 00:00:00 2001 From: yihong1120 Date: Sun, 30 Jun 2024 00:38:23 +0800 Subject: [PATCH] Debug: set default dataset path --- examples/YOLOv8-Evaluation/evaluate_yolov8.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/YOLOv8-Evaluation/evaluate_yolov8.py b/examples/YOLOv8-Evaluation/evaluate_yolov8.py index 541c32b..e5b0a46 100644 --- 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: """