部署网络缺少Softmax层

Question

我尝试将预训练模型（VGG 19）用于DIGITS，但是我遇到了这个错误。

  

错误：您的部署网络缺少Softmax层！阅读   自定义网络的文档和/或查看标准网络   例如

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

我看了this和this尝试修改最后一层但我也遇到了错误。如何根据新数据集修改图层？

我尝试修改最后一层，但我收到了错误

  

错误：图层'softmax'在TRAIN阶段引用底部'fc8'，但此blob不包含在该阶段。请考虑使用include指令来限制该层的范围。

Answer 1

您遇到了问题，因为您正试图上传＆＃34; train / val＆＃34;网络，当你真的需要上传一个＆＃34;一体机＆＃34;网络。不幸的是，我们不能很好地记录这一点。我已创建an RFE以提醒我们改进文档。

尝试调整网络中的最后一层，如下所示：https://github.com/NVIDIA/DIGITS/blob/v4.0.0/digits/standard-networks/caffe/lenet.prototxt#L162-L184

有关详细信息，here是我建议将Caffe的示例网络更新为一体化网络，here是我更新默认DIGITS网络的方式成为一体化的网。