set --image_dir to be required (#13896)

2024-09-23 14:17:44 +08:00 · 2024-09-23 14:17:44 +08:00 · 3cc4ae9f37
parent f214be5912
commit 3cc4ae9f37
1 changed files with 1 additions and 1 deletions
--- 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)
+    parser.add_argument("--image_dir", type=str, required=True)
    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)