Update rec_infer_from_crop_imgs.py

2024-04-22 00:08:36 +08:00
parent 55823256ec
commit 9e2d4347b1
1 changed files with 9 additions and 3 deletions
--- a/src/rec_infer_from_crop_imgs.py
+++ b/src/rec_infer_from_crop_imgs.py
@@ -11,10 +11,16 @@ if __name__ == '__main__':
    os.chdir(Path(__file__).resolve().parent)
    parser = argparse.ArgumentParser()
    parser.add_argument(
-        '-img', 
+        '-img_dir', 
        type=str, 
-        required=True,
-        help='path to the input image'
+        help='path to the input image',
+        default='./detect_results/subimages'
+    )
+    parser.add_argument(
+        '-output_dir', 
+        type=str, 
+        help='path to the output dir',
+        default='./rec_results'
    )
    parser.add_argument(
        '--inference-mode',