部署caffe回归模型

Question

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

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

Answer 1

对于deploy，您只需要丢弃丢失层，在您的例子中是"EuclideanLoss"层。你的网络的输出就是你给损失层提供的"bottom"。在

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