diff --git a/training/src/infer_model_cli.py b/training/src/infer_model_cli.py new file mode 100644 index 0000000..4fa93d1 --- /dev/null +++ b/training/src/infer_model_cli.py @@ -0,0 +1,15 @@ +from inference import inference_model +import argparse + + +def parse_args(): + parser = argparse.ArgumentParser() + parser.add_argument("--model_path", type=str, required=True) + parser.add_argument("--data_path", type=str, required=True) + parser.add_argument("--ann_save_path", type=str, required=True) + parser.add_argument("--conf_thresh", type=float, default=0.2) + return parser.parse_args() + +if __name__ == "__main__": + args = parse_args() + inference_model(args.model_path, args.data_path, args.ann_save_path) \ No newline at end of file