deep-learning - 部署网络缺少 Softmax 层

Question

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

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

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

我读了这个，这试图修改最后一层，但我也得到了错误。如何根据新数据集修改图层？

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

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

Answer 1

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

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

有关更多信息，以下是我建议将 Caffe 的示例网络更新为一体式网络的方法，以及我如何将默认 DIGITS 网络更新为一体式网络。