| 12 | |
| 13 | |
| 14 | def make_parser(): |
| 15 | parser = argparse.ArgumentParser("YOLOX onnx deploy") |
| 16 | parser.add_argument( |
| 17 | "--output-name", type=str, default="bytetrack_s.onnx", help="output name of models" |
| 18 | ) |
| 19 | parser.add_argument( |
| 20 | "--input", default="images", type=str, help="input node name of onnx model" |
| 21 | ) |
| 22 | parser.add_argument( |
| 23 | "--output", default="output", type=str, help="output node name of onnx model" |
| 24 | ) |
| 25 | parser.add_argument( |
| 26 | "-o", "--opset", default=11, type=int, help="onnx opset version" |
| 27 | ) |
| 28 | parser.add_argument("--no-onnxsim", action="store_true", help="use onnxsim or not") |
| 29 | parser.add_argument( |
| 30 | "-f", |
| 31 | "--exp_file", |
| 32 | default=None, |
| 33 | type=str, |
| 34 | help="expriment description file", |
| 35 | ) |
| 36 | parser.add_argument("-expn", "--experiment-name", type=str, default=None) |
| 37 | parser.add_argument("-n", "--name", type=str, default=None, help="model name") |
| 38 | parser.add_argument("-c", "--ckpt", default=None, type=str, help="ckpt path") |
| 39 | parser.add_argument( |
| 40 | "opts", |
| 41 | help="Modify config options using the command-line", |
| 42 | default=None, |
| 43 | nargs=argparse.REMAINDER, |
| 44 | ) |
| 45 | |
| 46 | return parser |
| 47 | |
| 48 | |
| 49 | @logger.catch |