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

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

Merged
tqchen merged 1 commit into
dmlc:masterfrom
tornadomeet:master
Dec 20, 2015
Merged

[RFC] add ignore-label for image segmentation#87
tqchen merged 1 commit into
dmlc:masterfrom
tornadomeet:master

Conversation

@tornadomeet

Copy link
Copy Markdown
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

tqchen commented Dec 19, 2015

Copy link
Copy Markdown
Member

@piiswrong can you review this?

@tqchen

tqchen commented Dec 19, 2015

Copy link
Copy Markdown
Member

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
Copy Markdown
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