Skip to content

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

@WarrenDream

Description

@WarrenDream

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions