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

How to train a multi-objective semantic segmentation data set #94

Closed
Atumyugi opened this issue Feb 20, 2024 · 2 comments
Closed

How to train a multi-objective semantic segmentation data set #94

Atumyugi opened this issue Feb 20, 2024 · 2 comments

Comments

@Atumyugi
Copy link

Different from the example data set, each image in my data set has multiple targets that need to be identified. The categories of these targets may not be consistent. Please follow the example and change the multiple targets to 0 255. Can it be trained? I only need the identification ability of SAM and do not need the classification ability. Or do I need to change the data set to a data set with only one target unit?
eg:
123

i need identifiy multi building

@Atumyugi
Copy link
Author

If i follow the train example and change the multiple targets to single 0 255. Can it be trained?

@WuJunde
Copy link
Collaborator

WuJunde commented Mar 6, 2024

We've supported multi-class segmentation (maximum 4 classes). Use it by setting -multimask_output to the number of classes favored. Also updated REFUGE example to two classes (optic disc & cup). Credit: @LJQCN101

@WuJunde WuJunde closed this as completed Mar 6, 2024
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