Error when converting from ONNX model to Tensorrt

Robotics & Edge Computing Jetson Systems Jetson AGX Xavier
, ,
1

I am trying to convert MobileSAM ONNX to tensorrt on Jetson AGX Xavier and I have been getting the following error which I am unable to resolve. Could you please guide me on how to solve?

[08/06/2023-01:11:33] [V] [TRT] ModelImporter.cpp:103: Parsing node: /Split [Split]
[08/06/2023-01:11:33] [V] [TRT] ModelImporter.cpp:119: Searching for input: /Expand_2_output_0
[08/06/2023-01:11:33] [V] [TRT] ModelImporter.cpp:119: Searching for input: /Constant_29_output_0
[08/06/2023-01:11:33] [V] [TRT] ModelImporter.cpp:125: /Split [Split] inputs: [/Expand_2_output_0 → ()], [/Constant_29_output_0 → (1)],
ERROR: onnx2trt_utils.cpp:188 In function convertAxis:
[8] Assertion failed: axis >= 0 && axis < nbDims
[08/06/2023-01:11:33] [E] Failed to parse onnx file
[08/06/2023-01:11:33] [E] Parsing model failed
[08/06/2023-01:11:33] [E] Engine creation failed
[08/06/2023-01:11:33] [E] Engine set up failed

CUDA version: 10.2
ONNX version : 1.14.0
Tensorrt version: 7.1.3.0
Jetpack version: 4.4.1

I have also attached my onnx file.

mobile_sam.onnx (15.7 MB)

6

Sorry for the late response, our team will do the investigation and provide suggestions soon. Thanks

8

There is no update from you for a period, assuming this is not an issue any more.
Hence we are closing this topic. If need further support, please open a new one.
Thanks

Dear @ckm26,
Could you share the used command. Also, is it possible to test with latest release (Jetpack 5.1.2)?