部署caffe回归模型

时间:2016-08-18 12:15:53

标签: python neural-network deep-learning caffe conv-neural-network

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

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

1 个答案:

答案 0 :(得分:2)

对于部署,您只需要丢弃损失层,在您的情况下是"EuclideanLoss"图层。您的网络输出是您输入损失图层的"bottom"

对于"SoftmaxWithLoss"图层(和"SigmoidCrossEntropy"),您需要替换损失图层,因为损失图层中包含一个额外的图层(出于计算原因)。 / p>