我用caffe
训练了一个回归网络。我在训练和测试阶段都使用“EuclideanLoss”层。我已经绘制了这些图,结果看起来很有希望。
现在我想部署模型并使用它。我知道如果使用 SoftmaxLoss,部署文件中的最后一层必须是 Softmax。在欧几里得损失的情况下,这应该是什么?
最佳答案
对于部署,您只需要丢弃损失层,在您的情况下是“EuclideanLoss”层。网络的输出是您输入损失层的“bottom”
。
对于“SoftmaxWithLoss”
层(和“SigmoidCrossEntropy”
),您需要替换损失层,因为损失层包含一个额外的其内部的层(出于计算原因)。
关于python - 部署caffe回归模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39017998/