123456789101112131415 |
- import os
- import argparse
- def main():
- args = parse_args()
- os.system('x2paddle --framework=onnx --model=' + args.onnx_model + ' --save_dir=' + args.output_dir)
- def parse_args():
- parser = argparse.ArgumentParser()
- parser.add_argument("--onnx_model", type=str, help="onnx model path")
- parser.add_argument("--output_dir", type=str, help="paddle model save dir")
- return parser.parse_args()
- if __name__ == "__main__":
- main()
|