Skip to content
This repository has been archived by the owner on Jul 1, 2024. It is now read-only.

YOLO&SAM onnx exporter #748

Open
yanzongs opened this issue May 7, 2024 · 2 comments
Open

YOLO&SAM onnx exporter #748

yanzongs opened this issue May 7, 2024 · 2 comments

Comments

@yanzongs
Copy link

yanzongs commented May 7, 2024

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?

@Greg7000
Copy link

Greg7000 commented May 8, 2024

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)

@yanzongs
Copy link
Author

yanzongs commented May 10, 2024

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants