Skip to content

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

Signed-off-by: tangy5 <yucheng.tang@vanderbilt.edu>
@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 <yucheng.tang@vanderbilt.edu>
@tangy5 tangy5 force-pushed the text_driven_segmentor branch from 4f98a2b to 6bb3d12 Compare April 5, 2023 04:55
Signed-off-by: tangy5 <yucheng.tang@vanderbilt.edu>
@tangy5 tangy5 force-pushed the text_driven_segmentor branch from e714c57 to 2b7b04b Compare April 5, 2023 05:04
Signed-off-by: tangy5 <yucheng.tang@vanderbilt.edu>
@tangy5 tangy5 force-pushed the text_driven_segmentor branch from 117002a to 9923686 Compare April 5, 2023 05:08
Signed-off-by: tangy5 <yucheng.tang@vanderbilt.edu>
@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