部署网络缺少一个Softmax层

我尝试使用pretrained模型（VGG 19）来DIGITS，但我得到了这个错误。部署网络缺少一个Softmax层

ERROR: Your deploy network is missing a Softmax layer! Read the documentation for custom networks and/or look at the standard networks for examples

我尝试使用只有两个类的数据集进行测试。

我读this和this尝试修改最后一层，但也得到错误。我如何根据新的数据集来修改图层？

我尝试修改的最后一层，我得到了错误

ERROR: Layer 'softmax' references bottom 'fc8' at the TRAIN stage however this blob is not included at that stage. Please consider using an include directive to limit the scope of this layer.

来源

2016-10-20 Video Analysis Deep Learning

你有，因为你想上传一个“火车/ VAL”网络出现了问题，当你确实需要上传“一体化”网络。不幸的是，我们没有很好地记录这一点。我创建了an RFE以提醒我们改进文档。

尝试调整最后一层网络中看起来是这样的：https://github.com/NVIDIA/DIGITS/blob/v4.0.0/digits/standard-networks/caffe/lenet.prototxt#L162-L184

欲了解更多信息，here是我是如何提出的更新来自Caffe的例子网络的所有功能于一身的渔网，并here是我如何将默认的DIGITS网络更新为一体化网络。

来源

2016-10-20 01:30:19

谢谢，但我得到错误，当我改变最后一层。 “错误：图层'softmax'在TRAIN阶段引用底部'fc8'，但该阶段不包含此blob，请考虑使用include指令来限制此图层的范围。我怎样才能改变最后一层？ –

如何更改最后一层？我怎样才能使用“一体化”？与微调模型相关的“一体式”？对于使用VGG 19层，我如何修改最后一层以便在仅有两个类的数据集中使用？ –

你可能会发现[这个答案]（http://stackoverflow.com/a/33773152/1714410）很有用：如何将“train”原型文件转换为“部署”原型文件。 – Shai

部署网络缺少一个Softmax层

回答

相关问题