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

Add text driven segmentor for dynamic outputs #6283

Draft
wants to merge 10 commits into
base: dev
Choose a base branch
from

Conversation

tangy5
Copy link
Contributor

@tangy5 tangy5 commented Apr 4, 2023

This PR adds a head controller and Universal model for promptable outputs.

Link to the issue: #6178

Users can choose and control the number and name of output channels from a multi-class segmentation model. This can enable incremental learning by adding new classes to an existing pre-trained model without catastrophic forgetting.

Working in progress: to add Universal Model class

@tangy5 tangy5 self-assigned this Apr 4, 2023
@tangy5 tangy5 changed the title Add text driven segmentor for promptable outputs Add text driven segmentor for dynamic outputs Apr 5, 2023
Signed-off-by: tangy5 <[email protected]>
@tangy5 tangy5 force-pushed the text_driven_segmentor branch from 4f98a2b to 6bb3d12 Compare April 5, 2023 04:55
Signed-off-by: tangy5 <[email protected]>
@tangy5 tangy5 force-pushed the text_driven_segmentor branch from e714c57 to 2b7b04b Compare April 5, 2023 05:04
Signed-off-by: tangy5 <[email protected]>
@tangy5 tangy5 force-pushed the text_driven_segmentor branch from 117002a to 9923686 Compare April 5, 2023 05:08
Signed-off-by: tangy5 <[email protected]>
@tangy5 tangy5 force-pushed the text_driven_segmentor branch from f9733ef to cb1402c Compare April 5, 2023 06:24
@tangy5
Copy link
Contributor Author

tangy5 commented Apr 13, 2023

Thanks. I will reformat this PR, and this PR might need further discussion.
As I'd like to create a dynamic segmentation head that can attach to most segmentation models.

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.

1 participant