diff --git a/ocr.py b/ocr.py index 8a09ffe483cddd30555be98617c46ea03af1c923..91e4343d13aba57c156d5501872ff01db46f89f5 100644 --- a/ocr.py +++ b/ocr.py @@ -15,3 +15,8 @@ rec_model_dir = "./inference_model/rec_v4" cls_model_dir = "./inference_model/ch_ppstructure_mobile_v2.0_SLANet_infer" result_dir = "./inference_results/v4" +# 创建目录 +os.makedirs(upload_dir, exist_ok=True) +os.makedirs(output_dir, exist_ok=True) +os.makedirs(result_dir, exist_ok=True) +