From cda3e1201ea50c46c312de0ef1c9c91199d31aab Mon Sep 17 00:00:00 2001 From: Wang Xin Date: Fri, 27 Sep 2024 15:58:49 +0800 Subject: [PATCH] fix hubserving run error (#13918) * fix hubserving run error * Update paddleocr.py --- paddleocr.py | 2 ++ tools/infer/utility.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/paddleocr.py b/paddleocr.py index 0f7dabd160..2dbefd566e 100644 --- a/paddleocr.py +++ b/paddleocr.py @@ -915,8 +915,10 @@ def main(): """ # for cmd args = parse_args(mMain=True) + logger.info("for usage help, please use `paddleocr --help`") image_dir = args.image_dir if is_link(image_dir): + os.remove("tmp.jpg") if os.path.exists("tmp.jpg") else None download_with_progressbar(image_dir, "tmp.jpg") image_file_list = ["tmp.jpg"] else: diff --git a/tools/infer/utility.py b/tools/infer/utility.py index a0ac59ff86..c40fc6b1b6 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -49,7 +49,7 @@ def init_args(): parser.add_argument("--gpu_id", type=int, default=0) # params for text detector - parser.add_argument("--image_dir", type=str, required=True) + parser.add_argument("--image_dir", type=str) parser.add_argument("--page_num", type=int, default=0) parser.add_argument("--det_algorithm", type=str, default="DB") parser.add_argument("--det_model_dir", type=str)