From 50758e201465028450d2dab484618a87ee2c647a Mon Sep 17 00:00:00 2001
From: wuzekai <3025054974@qq.com>
Date: Tue, 8 Jul 2025 05:27:28 +0000
Subject: [PATCH] Update test.py

---
 test.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/test.py b/test.py
index 06b40e4..daa98b3 100644
--- a/test.py
+++ b/test.py
@@ -34,4 +34,22 @@ command_v2 = [
     "--draw_img_save_dir", save_dir_v2,
     # "--convert_to_gray",
     # "--enable_preprocess",
-]
\ No newline at end of file
+]
+
+s|u|b|p|r|o|c|e|s|s.run(command_v2, check=True)
+print(f"[INFO] Model v2 results saved in: {save_dir_v2}")
+
+# 调用 v4 模型
+command_v4 = [
+    "python",
+    "tools/infer/predict_system.py",
+    "--image_dir", image_dir,
+    "--det_model_dir", det_model_v4,
+    "--rec_model_dir", rec_model_v4,
+    "--cls_model_dir", cls_model,
+    "--draw_img_save_dir", save_dir_v4,
+    # "--convert_to_gray",
+    # "--enable_preprocess",
+]
+s|u|b|p|r|o|c|e|s|s.run(command_v4, check=True)
+print(f"[INFO] Model v4 results saved in: {save_dir_v4}")
\ No newline at end of file
-- 
2.22.0