[A] Add testing benchmark

DocsaidLab · Mar 26, 2024 · 33768bd · 33768bd
1 parent 36fe218
commit 33768bd
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
@@ -0,0 +1,15 @@
+import docsaidkit as D
+from docclassifier import DocClassifier
+
+DIR = D.get_curdir(__file__)
+
+classifier = DocClassifier(
+    register_root='/home/shayne/workspace/DocClassifier/data/private/RegisterCard',
+    model_cfg='20240326'
+)
+
+fs = D.get_files(DIR.parent / 'benchmark' /
+                 'jpn_driver_color_benchmark_dataset', suffix=['.jpg'])
+for f in fs:
+    img = D.imread(f)
+    print(f' 定義為：{f.parent.name}，預測為：{classifier(img)[0]}')