Update test.py

test.py

@@ -14,3 +14,24 @@ cls_model = "./inference_model/ch_ppstructure_mobile_v2.0_SLANet_infer"
 # 定义图像保存路径
 save_dir_v2 = "./inference_results/v2"
 save_dir_v4 = "./inference_results/v4"
+
+# 检查路径是否存在
+if not os.path.exists(image_dir):
+    raise FileNotFoundError(f"Image directory not found: {image_dir}")
+if not os.path.exists(det_model_v2) or not os.path.exists(rec_model_v2):
+    raise FileNotFoundError("det_v2 or rec_v2 model directory not found")
+if not os.path.exists(det_model_v4) or not os.path.exists(rec_model_v4):
+    raise FileNotFoundError("det_v4 or rec_v4 model directory not found")
+
+# 调用 v2 模型
+command_v2 = [
+    "python",
+    "tools/infer/predict_system.py",
+    "--image_dir", image_dir,
+    "--det_model_dir", det_model_v2,
+    "--rec_model_dir", rec_model_v2,
+    "--cls_model_dir", cls_model,
+    "--draw_img_save_dir", save_dir_v2,
+    # "--convert_to_gray",
+    # "--enable_preprocess",
+]
\ No newline at end of file