IDEA-Research / Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything
https://arxiv.org/abs/2401.14159
Apache License 2.0
15.09k stars 1.39k forks source link

How to using the text prompt with SAM? #92

Open Lufffya opened 1 year ago

Lufffya commented 1 year ago

image

在这里,我指定了湖,但仍然框中了熊,是我用错了吗?

请问作者是如何把 text prompt 和 SAM 结合起来的,SAM 模型好像只支持 boxes 和 points

rentainhe commented 1 year ago

你可以需要修改一下task type~ 这里可以让 @tuofeilunhifi 来解答一下~

GroundingDINO可以结合text出boxes,然后再利用SAM的能力,就可以出Box + Mask了