CODEWITHSUNDEEP
pythonneural-networkdeep-learningcaffeconv-neural-network
Cassie
Cassie

Reputation: 372

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?

Upvotes: 2

Views: 621

Answers (1)

Shai
Shai

Reputation: 114816

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).

Upvotes: 2

Related Questions