Support export efficientvit-sam models to ONNX

mit-han-lab / efficientvit

EfficientViT is a new family of vision models for efficient high-resolution vision.

Apache License 2.0

1.6k stars 142 forks source link

Support export efficientvit-sam models to ONNX #27

Open CVHub520 opened 9 months ago

CVHub520 commented 9 months ago

Recognizing the remarkable open-source efforts of the author, who has contributed immensely, it's worth noting that EfficiencyViT currently doesn't provide scripts to export the sam encoder and decoder as ONNX formats, here I'm offering a simple way to support it. Feel free to grab it if you need it. Additionally, I have integrated this model into the X-AnyLabeling tool, and the results are truly impressive. We encourage everyone to give it a try enthusiastically.

han-cai commented 9 months ago

Hello,

Thanks for sharing your script and your impressive work!

Do you mind if I reference your script in our README?

Best, Han

CVHub520 commented 9 months ago

Hello Han,

Thank you for reaching out and for your kind words! I'm glad to hear that you find the script useful and would like to reference it in your README. You are more than welcome to do so. If you have any specific information or details you'd like to include in the reference, feel free to let me know, and I'll be happy to assist further.

Best regards, CVHub