Update rec_infer_from_crop_imgs.py

This commit is contained in:
TonyLee1256
2024-04-22 00:08:36 +08:00
committed by GitHub
parent 55823256ec
commit 9e2d4347b1

View File

@@ -11,10 +11,16 @@ if __name__ == '__main__':
os.chdir(Path(__file__).resolve().parent) os.chdir(Path(__file__).resolve().parent)
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
'-img', '-img_dir',
type=str, 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( parser.add_argument(
'--inference-mode', '--inference-mode',