Custom SAM Model

[ajaidevanath]

How to use custom SAM model trained on custom medical segmentation data?

[piyushw0203]

heyy, were you able to figure it out?

[luca-medeiros]

you can pass the model path to ckpt_path argument here

lang-segment-anything/lang_sam/lang_sam.py

Line 9 in cfb29a6

def __init__(self, sam_type="sam2.1_hiera_small", ckpt_path: str | None = None):

Not tested with SAM2.1 release.

[ajaidevanath]

Thank you Lucas, Wanted to check where do we get the checkpoint for the "sam2.1_hiera_small" model

[luca-medeiros]

@ajaidevanath
For the original models from Meta you don't need to download the models lang-sam does it internally. You only need to use ckpt_path if you are using a custom model.

[ajaidevanath]

ok got it , Thanks for the help :-)