bowang-lab / MedSAM

Segment Anything in Medical Images
https://www.nature.com/articles/s41467-024-44824-z
Apache License 2.0
3.02k stars 419 forks source link

convert onnx #335

Open lucky-asia opened 2 weeks ago

lucky-asia commented 2 weeks ago

After training with my own dataset, can I export the saved weights to ONNX? How to convert?

JunMa11 commented 3 days ago

Hi,

MedSAM doesn't support ONNX but welcome to try this one, which is the 1st place in the cvpr competition.

https://github.com/hieplpvip/medficientsam