Hi mingj2021,

I had a problem when using the export_sam_model function in demo.py script:
If the "orig_im_size" input in dummy_inputs is commented (as default), I cannot export the onnx model.
Got the following error:
TypeError: forward() missing 1 required positional argument: 'orig_im_size'

If I add the "orig_im_size" input, export can be success but cannot further transfer to .engine model.

Any hints to solve this issue?

I had tried both sam_vit_b_01ec64.pth and sam_vit_l_0b3195.pth
With TensorRT-8.5.3.1