I try to export sam mask_decoder.onnx to tensorrt，but got error

[zhishao]

Has anyone successfully exported the model to TensorRT?

[zhishao]

[08/21/2025-16:37:00] [TRT] [E] IBuilder::buildSerializedNetwork: Error Code 4: API Usage Error (Optimization profile 0 is missing values for shape input tensor orig_im_size.)

[duxuan11]

            min_shape = (512, 512)
            opt_shape = (1024, 1024)
            max_shape = (2048, 2048)
            profile.set_shape_input(name, min_shape , opt_shape, max_shape )