Skip to content

Added support for MobileSAM for snappier Gradio experience#100

Open
JunkyByte wants to merge 4 commits intogaomingqi:masterfrom
JunkyByte:master
Open

Added support for MobileSAM for snappier Gradio experience#100
JunkyByte wants to merge 4 commits intogaomingqi:masterfrom
JunkyByte:master

Conversation

@JunkyByte
Copy link

I have added support to MobileSAM which can be used to experience a snappier Gradio interface as it runs faster on cpu than the original SAM.

During tracking it does not impact much, especially on CUDA.

I have removed args.device = "cuda:3" in app.py as it prevents from correctly selecting devices when running the script.

This adds MobileSAM as a requirement.

@yclicc
Copy link

yclicc commented Sep 8, 2023

This is great, but there are two bugs. 1, in your requirement.txt the new line that installs MobileSAM shouldn't have "pip install" in it. 2. in app.py you haven't set the SAM_checkpoint_url_dict value for the new vit_t, so it doesn't know where to download it from (maybe solve by adding an instruction to manually download it).

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