Skip to content
This repository has been archived by the owner on Aug 11, 2020. It is now read-only.

[RFC] add ignore-label for image segmentation #87

Merged
merged 1 commit into from
Dec 20, 2015
Merged

[RFC] add ignore-label for image segmentation #87

merged 1 commit into from
Dec 20, 2015

Conversation

tornadomeet
Copy link
Contributor

hello,
i have refactored the example fcn-xs in https://github.com/tornadomeet/mxnet/tree/seg/example/fcn-xs, and in image segmentation, the label image is like this http://host.robots.ox.ac.uk/pascal/VOC/voc2012/segexamples/index.html
as you can see, the index 255 corresponds to 'void' or unlabelled for ambiguity in the edge of object, so i add one SoftmaxGrad function for this problem, i just follow the caffe implentation https://github.com/longjon/caffe/blob/future/src/caffe/layers/softmax_loss_layer.cpp#L103
and we can debat this, maybe there is a better way do it.
thanks~

@tqchen
Copy link
Member

tqchen commented Dec 19, 2015

@piiswrong can you review this?

@tqchen
Copy link
Member

tqchen commented Dec 19, 2015

Please make a consistent CPU implementation as well, and document this behavior carefully. As this is not same as normal SoftmaxGrad, but something like special unknown label

@piiswrong
Copy link
Member

LGTM

tqchen added a commit that referenced this pull request Dec 20, 2015
[RFC] add ignore-label for image segmentation
@tqchen tqchen merged commit 47521c6 into dmlc:master Dec 20, 2015
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants