search

huggingface / controlnet_aux

Apache License 2.0
398 stars 86 forks source link

Add mobileSAM in SamDetector #63

Closed jinwonkim93 closed 1 year ago

jinwonkim93 commented 1 year ago

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 commented 1 year ago

Very cool addition!