部署 caffe 回歸模型

Question

我用caffe訓練了一個回歸網絡。 我在訓練和測試階段都使用"EuclideanLoss"層。 我已經繪制了這些，結果看起來很有希望。

現在我想部署模型並使用它。 我知道如果使用SoftmaxLoss ，deploy文件中最后一層一定是Softmax 。 在Euclidean loss的情況下這應該是什么？

Answer 1

對於部署，您只需要丟棄損失層，在您的情況下是"EuclideanLoss"層。 您的網絡的輸出是您輸入損失層的"bottom" 。

對於"SoftmaxWithLoss"層（和"SigmoidCrossEntropy" ），您需要替換損失層，因為損失層在其中包含一個額外的層（出於計算原因）。