An-619/FastSAM · Some suggestions for the demo!

Jun 22, 2023

Very cool demo! Just a few suggestions:

(1) I was getting an OOM error whenever I tried it -- you might need to upgrade to GPU?
(2) Can you set cache_examples = True to make it faster to see the results on the provided examples?
(3) Perhaps the gr.AnnotatedImage component would provide for a nicer output: https://gradio.app/docs/#annotatedimage

Also, if you cite your paper http://arxiv.org/abs/2306.12156 in the README, it will show up in the associated paper page: https://huggingface.co/papers/2306.12156

An-619

Owner Jun 22, 2023

Thanks for your suggestion. I have upgraded to GPU and to show an example as "cache_examples = True". The queue and more cache examples will come soon.

ukaprch

Oct 30, 2023

Currently working with this demo (locally) adding some features (i.e. box prompt, mask prompt, show segment or show mask generated, load one input file and use for all prompts, etc.
Your demo and working code is very nice and much appreciated.