Skip to content

Add mobileSAM in SamDetector#63

Merged
patrickvonplaten merged 3 commits intohuggingface:masterfrom
jinwonkim93:feat/mobileSAM
Jul 3, 2023
Merged

Add mobileSAM in SamDetector#63
patrickvonplaten merged 3 commits intohuggingface:masterfrom
jinwonkim93:feat/mobileSAM

Conversation

@jinwonkim93
Copy link
Contributor

@jinwonkim93 jinwonkim93 commented Jun 30, 2023

Add new model MobileSAM in SamDetector.

It works well and does not break SamDetector.
It is faster than sam_b

official code from https://github.com/ChaoningZhang/MobileSAM

from controlnet_aux import SamDetector
mobile_sam = SamDetector.from_pretrained("dhkim2810/MobileSAM", model_type="vit_t", filename="mobile_sam.pt")

dog

output

@patrickvonplaten
Copy link
Contributor

Very cool addition!

@patrickvonplaten patrickvonplaten merged commit 1e6bdc0 into huggingface:master Jul 3, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants