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

Allow segmentation color map to be specified explicitly #993

Merged

Conversation

gheinrich
Copy link
Contributor

Add form fields to allow users to provide separate text file with color/class mappings.
This is useful when working with datasets that have label images in RGB.

Depends on #961

@gheinrich gheinrich force-pushed the dev/explicit-segmentation-color-map branch from caebfb0 to 0343f16 Compare August 30, 2016 12:16
@lukeyeager
Copy link
Member

I'm not sure how to test this - is there a public dataset that uses RGB labels like this?

When it's convenient, can you rebase this now that #961 is merged (just for clarify / ease of review)?

@gheinrich gheinrich force-pushed the dev/explicit-segmentation-color-map branch from 0343f16 to 3110fef Compare August 30, 2016 20:06
This is useful when working with image labels in RGB format.
A separate text file can be provided to specify color/class mappings.

Conflicts:
	digits/extensions/data/imageSegmentation/forms.py
@gheinrich gheinrich force-pushed the dev/explicit-segmentation-color-map branch from 3110fef to 278d24b Compare August 30, 2016 20:18
@gheinrich
Copy link
Contributor Author

Thanks. This change is now rebased on the tip of the master branch. An example of a segmentation dataset with RGB labels is SYNTHIA. The color/class mappings for this dataset are given in the dataset's README.txt and may be provided to DIGITS through a text file that contains:

0   0   0
128 128 128
128 0   0
128 64  128
0   0   192
64  64  128
128 128 0
192 192 128
64  0   128
192 128 128
64  64  0
0   128 192

Ideally you should update the FCN prototxt to reduce the number of outputs to 12 (since there are 12 classes in this dataset) though I found it that it wasn't necessary to get the network to converge.

@lukeyeager lukeyeager self-assigned this Sep 6, 2016
@lukeyeager
Copy link
Member

LGTM

@lukeyeager lukeyeager merged commit 096f90a into NVIDIA:master Sep 6, 2016
@gheinrich gheinrich deleted the dev/explicit-segmentation-color-map branch November 30, 2016 16:50
SlipknotTN pushed a commit to cynnyx/DIGITS that referenced this pull request Mar 30, 2017
…n-color-map

 	Allow segmentation color map to be specified explicitly
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants