python - 部署caffe回归模型

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

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

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

最佳答案

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

对于“SoftmaxWithLoss”层(和“SigmoidCrossEntropy”),您需要替换损失层,因为损失层包含一个额外的其内部的层(出于计算原因)。

关于python - 部署caffe回归模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/39017998/

相关文章:

python - 如何匹配字典中的两个值并合并结果

python - 将字符串转换为二元组列表

tensorflow - Inception 模型有两个 softmax 输出吗?

javascript - Uncaught ReferenceError : PythonShell is not defined - PythonShell for Nodejs and Electron

python - 当字典不存在时设置默认值

python - Keras 中 n 维函数的导数

machine-learning - PyBrain buildNetwork 与 FeedForwardNetwork

machine-learning - Caffe - 其他 N 个最后迭代的平均准确度

tensorflow - 如何在非 NVIDIA 设置上加速深度学习?

python - 无法使用 ResNet50 在 Keras 中加载权重以进行微调