luca-medeiros/lang-segment-anything

Load samhq model?

Closed this issue · 3 comments

Hey
https://github.com/luca-medeiros/lang-segment-anything#usage
here you can find custom checkpoint usage. havent test your case tho

image
How to soulve the problem??

Hey, I think I found the issue.

sam = sam_model_registry[self.sam_type](ckpt_path)

I believe here should be:
sam = sam_model_registry[self.sam_type](checkpoint=ckpt_path)

Mind trying and writing a PR if it works?