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

[WIP] Multiclass Support #7

Open
wants to merge 12 commits into
base: master
Choose a base branch
from
Open

[WIP] Multiclass Support #7

wants to merge 12 commits into from

Conversation

khdlr
Copy link
Collaborator

@khdlr khdlr commented Sep 13, 2021

This PR consists of Aleks' works on multiclass segmentation.

It should be ready to merge once we're in a place where recalculating everything is viable.

targ_al and others added 12 commits June 30, 2021 17:19
metrics.py, models and utils/plot_info.py were changed to work with more than one class (multiclass).
ToDo:
	- fix metrics to work with multiclass (Konrad)
	- make number of classes flexible (based on mask) (Aleks)
The user can now see the percentage of specific classes (features) which appear in the data.
ToDo: Right now hard coded class names in prepare_data.py. Next step: read feature labels from config.
Added config option to prepare_data.py to get the labels from config.yml.
Changed variable names from classes and labels to "features" to make it (hopefully) less confusing.
Added two asserts regarding problems that might appear.
- Added and already tested three warnings.
- config works now with a dictionary instead of array.
	example: data_classes: {0 : "background", 1 : "slope", 2 : "lake", 3 : "class3", 7 : "class7", 8 : "class8"}
@khdlr khdlr mentioned this pull request Sep 13, 2021
@khdlr khdlr linked an issue Sep 13, 2021 that may be closed by this pull request
@khdlr khdlr linked an issue Sep 13, 2021 that may be closed by this pull request
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.

Multiclass Support cap metrics output to fewer digits
1 participant