nanosam-cpp icon indicating copy to clipboard operation
nanosam-cpp copied to clipboard

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

Open WarrenDream opened this issue 2 years ago • 2 comments

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]?

WarrenDream avatar Jan 29 '24 08:01 WarrenDream

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.

spacewalk01 avatar Feb 01 '24 02:02 spacewalk01

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

hamdiboukamcha avatar Oct 28 '24 15:10 hamdiboukamcha