modify softmaxwithloss layer

[Yue Liu]

Hello,

I'm new to caffe. I wonder if it is possible to modify the src of SoftmaxWithLoss layer to the following setting:

when doing pixel wise classification and evaluating SoftmaxWithLoss, we can assign a weight to each pixel loss, (because for each image, the loss is the sum of all pixel loss), where pixels from dominant class have less impact to the loss function. In an unbalanced setting, e.g. in a label image, the most of the pixels are black, I would like to prevent the network producing all 0 by doing the modification for the loss function mentioned above, because this could be a local minimum.  I wonder if it is possible.

Thank you.

Yue