efficientvit icon indicating copy to clipboard operation
efficientvit copied to clipboard

ONNX only supports point prompts and box prompts. How to segment everything with onnxruntime?

Open panp4n opened this issue 1 year ago • 2 comments

Hi, ONNX only supports point prompts and box prompts. How to segment everything with Onnxruntime or TensorRT? By the way, is there a plan to add C++ inference demo? Thank you!

panp4n avatar May 08 '24 06:05 panp4n

We have supported segmenting everything with Onnxruntime/TensorRT. Please check the following README file: https://github.com/mit-han-lab/efficientvit/blob/master/demo/sam/README.md

han-cai avatar May 08 '24 14:05 han-cai

Thanks for your reply. I have read this document and the codes. It uses the api of SAM to segment everything. If I code with C++, how could I achieve this task?

panp4n avatar May 16 '24 06:05 panp4n