onnx2paddle.py 449 B

123456789101112131415
  1. import os
  2. import argparse
  3. def main():
  4. args = parse_args()
  5. os.system('x2paddle --framework=onnx --model=' + args.onnx_model + ' --save_dir=' + args.output_dir)
  6. def parse_args():
  7. parser = argparse.ArgumentParser()
  8. parser.add_argument("--onnx_model", type=str, help="onnx model path")
  9. parser.add_argument("--output_dir", type=str, help="paddle model save dir")
  10. return parser.parse_args()
  11. if __name__ == "__main__":
  12. main()