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()