部署网络缺少 Softmax 层

Question

我尝试将预训练模型 (VGG 19) 用于 DIGITS，但出现此错误。

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

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

我阅读了this 和this 尝试修改最后一层，但也出现错误。如何根据新数据集修改图层？

我尝试修改最后一层，但出现错误

错误：层“softmax”在 TRAIN 阶段引用了底部“fc8”，但该 blob 不包含在该阶段。请考虑使用包含指令来限制这一层的范围。

Answer 1

您遇到了问题，因为您在真正需要上传“一体式”网络时尝试上传“train/val”网络。不幸的是，我们没有很好地记录这一点。我创建了an RFE 来提醒我们改进文档。

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

如需更多信息，here 是我建议将 Caffe 的示例网络更新为一体式网络的方法，here 是我将默认 DIGITS 网络更新为一体式网络的方法。