Weighted Softmax Loss

Question

Weighted Softmax Loss

atanas1054 opened this issue 7 years ago · comments

Hello,

I want to use just the weighted softmax layer in SegNet without actually using the SegNet architecture. How can I just take the weighted softmax layer and compile it on the latest caffe version?

Nathan Ing · Answer 1 · Fri May 12 2017 02:34:00 GMT+0800 (China Standard Time)

Try to copy the files src/caffe/layers/softmax_loss_layer.cpp and .cu and include/caffe/layers/softmax_loss_layer.hpp

Also check src/caffe/proto/caffe.proto and see if the LossParameter "class_weighting" is there.

Hope it works.

atanas1054 · Answer 2 · Fri May 12 2017 03:48:00 GMT+0800 (China Standard Time)

Hey,
There is no include/caffe/layers/softmax_loss_layer.hpp

Nathan Ing · Answer 3 · Fri May 12 2017 06:05:28 GMT+0800 (China Standard Time)

This repo is based on an old version of caffe.

the repo here: https://github.com/TimoSaemann/caffe-segnet-cudnn5 has the layers factored out into individual header files. (see there's no "vision_layers.hpp", "loss_layers.hpp", etc.). I think this is how the most updated caffe does it still.

https://github.com/TimoSaemann/caffe-segnet-cudnn5/blob/master/include/caffe/layers/softmax_loss_layer.hpp

atanas1054 · Answer 4 · Fri May 12 2017 17:52:33 GMT+0800 (China Standard Time)

It's working now. Thank you.