SAM_TensorRT
SAM_TensorRT copied to clipboard
This is the code to implement Segment Anything (SAM) using TensorRT(C++).
In [MobileSAM](https://github.com/ChaoningZhang/MobileSAM), there is only one weight file named mobile_sam.pt, how to get encoder and decoder weigt separately? Could you share the script for that?
Nice work! In the quest for faster speeds, is the framework applicable to Nano SAM ?
thanks for your contribution. is this repo still suitable for edgeSAM?
SAM HQ?
Hi, thank you for this code. Can it support SAM HQ? https://github.com/SysCV/sam-hq
Hi, thank you for your efforts. I am currently testing the code using the "dog.jpg" example, but the results are not accurate. Initially, I drew a bounding box around the...
init enegien sucess,but forward error: kERROR: 3: [executionContext.cpp::nvinfer1::rt::ShapeMachineContext::resolveSlots::2509] Error Code 3: API Usage Error (Parameter check failed at: executionContext.cpp::nvinfer1::rt::ShapeMachineContext::resolveSlots::2509, condition: allInputDimensionsSpecified(routine) )
1) /usr/local/TensorRT-8.6.1.6//bin/trtexec --onnx=./models/mobile_sam_encoder.onnx --saveEngine=./models/mobile_sam_encoder.engine --memPoolSize=workspace:10240 2) TensorRT 8.6 change onnx to trt error log: &&&& RUNNING TensorRT.trtexec [TensorRT v8601] # /usr/local/TensorRT-8.6.1.6//bin/trtexec --onnx=./models/mobile_sam_encoder.onnx --saveEngine=./models/mobile_sam_encoder.engine --memPoolSize=workspace:10240 [11/03/2023-16:35:20] [I] === Model Options ===...
Does this algorithm process 1024x1024 ,no matter what size picture it processes ,Is the output image also that large?
RT,do you modify some code about mobiesam when you convert encoder.onnx?my converted encoder.onnx"s accuracy is not as good as you