Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Which prompt use to generate the masks? #9

Open
JooudDoo opened this issue Jan 24, 2024 · 3 comments
Open

Which prompt use to generate the masks? #9

JooudDoo opened this issue Jan 24, 2024 · 3 comments

Comments

@JooudDoo
Copy link

Is there some kind of template to explain to the model that a mask is needed?
Also, the model is very unlikely (almost never) to give multiple bboxes for the same class when requesting something like: "Detect XXX. Please include several object locations". Most often the model just gives a bbox that combines all matching objects into one, but I would like them to be separate.

@JooudDoo JooudDoo changed the title Which prompt to use to generate the masks? Which prompt use to generate the masks? Jan 24, 2024
@waxnkw
Copy link
Collaborator

waxnkw commented Jan 25, 2024

Thanks for your question! The "Detect XXX" prompt is mainly trained on the single object detection scenarios. A way to include multiple similar object is to use "Can you describe the image and detect objects?". However, this way is only for the whole image description but weak in referring the given objects. If you want to achieve this, I think a further fine-tuning on such kind of data is required. We are sorry about the incapability now.

@JooudDoo
Copy link
Author

Thanks for the reply. Is there a prompt that will tell the model that I need a mask?

@waxnkw
Copy link
Collaborator

waxnkw commented Jan 31, 2024

If there is a box, there will be a corresponding mask. However, I filter out some low-quality masks by using a predicted iou_thres. You can modify the iou_thres to 0 at

temperature=0.75, top_p=0.7, top_k=5, boxes=None, boxes_seq=None, iou_thres=0.3):
.

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

No branches or pull requests

2 participants