Update test.py

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