Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

infer result in mCpuBuffers[6] ? is not mCpuBuffers[5] ? #3

Open
WarrenDream opened this issue Jan 29, 2024 · 2 comments
Open

infer result in mCpuBuffers[6] ? is not mCpuBuffers[5] ? #3

WarrenDream opened this issue Jan 29, 2024 · 2 comments

Comments

@WarrenDream
Copy link

in trt_module.cpp getOutput()function "
memcpy(lowResolutionMasks, mCpuBuffers[5], mBufferBindingBytes[5]);
memcpy(iouPrediction, mCpuBuffers[6], mBufferBindingBytes[6]); " in nanosam.cpp set decoder: "mMaskDecoder = new TRTModule(decoderPath, { "image_embeddings", "point_coords", "point_labels", "mask_input", "has_mask_input" }, { "iou_predictions", "low_res_masks" }, true, false);" the infer result iou_predictions in mCpuBuffers[6]? is not mCpuBuffers[5]?

@spacewalk01
Copy link
Owner

spacewalk01 commented Feb 1, 2024

When I used trtexec for generating the engine from onnx model, somehow, the order of outputs in decoder were changed. So, I set it accordingly But I will debug it when I have a time.

@hamdiboukamcha
Copy link

@spacewalk01 @WarrenDream Please, can anyone provide the service and share both ONNX files? Thank you in advance.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants