Merge pull request #46 from ArdalanM/patch-1

fix cases when some categories are empty
2025-02-14 17:06:23 +08:00
parent f0f1972c6f 671a3f88c9
commit b5ecad3e34
1 changed files with 1 additions and 1 deletions
--- a/OCRBench_v2/eval_scripts/eval.py
+++ b/OCRBench_v2/eval_scripts/eval.py
@@ -359,7 +359,7 @@ def process_predictions(input_path, output_path):
                total_len += 1
                mean_score += item["score"]
        
-        mean_score = mean_score / total_len
+        mean_score = mean_score / total_len if total_len > 0 else 0
        print(f"Task {task_name}, total instructions: {total_len}, average score: {mean_score:.3f}\n")

    with open(output_path, 'w', encoding='utf-8') as file: