user4039874 user4039874 - 3 months ago 93
Python Question

Deploy caffe regression model

I have trained a regression network with

caffe
. I use
"EuclideanLoss"
layer in both the train and test phase. I have plotted these and the results look promising.

Now I want to deploy the model and use it. I know that if
SoftmaxLoss
is used, the final layer must be
Softmax
in the deploy file. What should this be in the case of
Euclidean loss
?

Answer

For deploy you only need to discard the loss layer, in your case the "EuclideanLoss" layer. The output of your net is the "bottom" you fed the loss layer.

For "SoftmaxWithLoss" layer (and "SigmoidCrossEntropy") you need to replace the loss layer, since the loss layer includes an extra layer inside it (for computational reasons).