(args)
| 71 | |
| 72 | |
| 73 | def onnx2trt(args): |
| 74 | from mmdeploy.backend.tensorrt import from_onnx |
| 75 | |
| 76 | onnx_name = f'{args.model_name}.onnx' |
| 77 | from_onnx( |
| 78 | onnx_name, |
| 79 | args.model_name, |
| 80 | dict( |
| 81 | input=dict( |
| 82 | min_shape=[1, 3, args.size, args.size], |
| 83 | opt_shape=[1, 3, args.size, args.size], |
| 84 | max_shape=[1, 3, args.size, args.size], |
| 85 | ) |
| 86 | ), |
| 87 | max_workspace_size=2 ** 30, |
| 88 | ) |
| 89 | |
| 90 | |
| 91 | def check(args, cfg): |