set --image_dir to be required (#13896)

PaddlePaddle · Sep 23, 2024 · 3cc4ae9 · 3cc4ae9
1 parent f214be5
commit 3cc4ae9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)