部署Caffe回归模型

小编典典

python

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

现在，我要部署模型并使用它。我知道如果SoftmaxLoss使用了，最后一层必须Softmax在部署文件中。在什么情况下该Euclidean loss怎么办？

阅读 196

2021-01-20

共1个答案

小编典典

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

对于"SoftmaxWithLoss"层（和"SigmoidCrossEntropy"），您需要替换
损耗层，因为损耗层在其中包括一个额外的层（出于计算原因）。

2021-01-20