segment-anything icon indicating copy to clipboard operation
segment-anything copied to clipboard

YOLO&SAM onnx exporter

Open yanzongs opened this issue 9 months ago • 2 comments

Hello, when I need to combine YOLO and SAM together, I need to export onnx together with the for boxes in the prompt encoder and mask decoder. Do you have any specific instructions to tell me how to do it?

yanzongs avatar May 07 '24 15:05 yanzongs

I recommend you take a look at:

https://github.com/AndreyGermanov/sam_onnx_full_export/blob/main/sam_onnx_export.ipynb or https://github.com/vietanhdev/samexporter (please note that on my side to manage to get it to work I need to perform local install instead of PyPi install)

Greg7000 avatar May 08 '24 20:05 Greg7000

I recommend you take a look at:

https://github.com/AndreyGermanov/sam_onnx_full_export/blob/main/sam_onnx_export.ipynb or https://github.com/vietanhdev/samexporter (please note that on my side to manage to get it to work I need to perform local install instead of PyPi install)

However, the tutorial you provided still converts prompt encoder and mask decoder together into decoder part, could you please just generate decoder

yanzongs avatar May 10 '24 02:05 yanzongs