tianrun-chen / SAM-Adapter-PyTorch

Adapting Meta AI's Segment Anything to Downstream Tasks with Adapters and Prompts
MIT License
1.06k stars 97 forks source link

support for prompts #45

Open Abhiram-kandiyana opened 1 year ago

Abhiram-kandiyana commented 1 year ago

Hi, Great work. Does your code support prompts, like point-prompts or box prompts which are in the original SAM

tianrun-chen commented 1 year ago

Greetings! This project primarily addresses traditional image segmentation scenarios, such as shadow detection without prompt input. As a result, we have not integrated a prompt encoder into the neural network. However, you have the option to incorporate the prompt encoder from SAM if you wish to exercise control over the segmentation process, while still utilizing our modified SAM image encoder and decoder. Feel free to enable the prompt encoder and leverage its capabilities alongside our adapted SAM components.