部署caffe回归模型

Question

我用 caffe 训练了一个回归网络。我在训练和测试阶段都使用 "EuclideanLoss" 层。我已经绘制了这些，结果看起来很有希望。

现在我想部署模型并使用它。我知道如果使用SoftmaxLoss，最后一层在deploy文件中肯定是Softmax。在 Euclidean loss 的情况下应该是什么？

Answer 1

对于部署，您只需要丢弃丢失层，在您的情况下为 "EuclideanLoss" 层。你的网络的输出是 "bottom" 你喂给损失层的。

对于 "SoftmaxWithLoss" 层（和 "SigmoidCrossEntropy"），您需要 替换 损失层，因为损失层内部包含一个额外的层（对于计算原因）。