search

luca-medeiros / lang-segment-anything

SAM with text prompt
Apache License 2.0
1.53k stars 167 forks source link

Custom Checkpoints #27

Closed KabirSubbiah closed 1 year ago

KabirSubbiah commented 1 year ago

Hi @luca-medeiros,

Great work! I was using the model and thought it would be useful if LangSAM() accepted a SAM ckpt as an argument. This would allow for other checkpoints like HQ_SAM to be included more easily. Also, for people who have downloaded the model and are working in Colab, this would free up memory.

Happy to help!

Thanks,

Kabir

luca-medeiros commented 1 year ago

Hey @KabirSubbiah thanks That sounds like a nice feature! Wanna write a pr?

KabirSubbiah commented 1 year ago

Done,

Let me know if any changes need to be made!