diff --git a/test.py b/test.py index 06b40e4adb6df804922db328a60c5f90da18fe77..daa98b3224537686dba53d1a92b6811ad57ecdd5 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