Fix OCRBench evaluate

open-compass · Mar 21, 2024 · 2c1a5ed · 2c1a5ed
1 parent fd3aec0
commit 2c1a5ed
Showing 1 changed file with 13 additions and 13 deletions.
diff --git a/vlmeval/evaluate/OCRBench.py b/vlmeval/evaluate/OCRBench.py
@@ -1,20 +1,20 @@
 from vlmeval.smp import *
 
-OCRBench_score = {
-    'Regular Text Recognition': 0,
-    'Irregular Text Recognition': 0,
-    'Artistic Text Recognition': 0,
-    'Handwriting Recognition': 0,
-    'Digit String Recognition': 0,
-    'Non-Semantic Text Recognition': 0,
-    'Scene Text-centric VQA': 0,
-    'Doc-oriented VQA': 0,
-    'Key Information Extraction': 0,
-    'Handwritten Mathematical Expression Recognition': 0
-}
-
 
 def OCRBench_eval(eval_file):
+    OCRBench_score = {
+        'Regular Text Recognition': 0,
+        'Irregular Text Recognition': 0,
+        'Artistic Text Recognition': 0,
+        'Handwriting Recognition': 0,
+        'Digit String Recognition': 0,
+        'Non-Semantic Text Recognition': 0,
+        'Scene Text-centric VQA': 0,
+        'Doc-oriented VQA': 0,
+        'Key Information Extraction': 0,
+        'Handwritten Mathematical Expression Recognition': 0
+    }
+
     logger = get_logger('Evaluation')
 
     data = load(eval_file)