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

标签 deep-learning caffe softmax nvidia-digits

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

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

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

我读到thisthis尝试修改最后一层,但我也收到错误。如何根据新数据集修改图层?

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

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.

最佳答案

您遇到问题是因为当您确实需要上传“all-in-one”网络时却尝试上传“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 网络更新为一体化网络的方法。

关于deep-learning - 部署网络缺少 Softmax 层,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40143901/

相关文章:

python - 在Keras中,如何对权重矩阵的每一行应用softmax函数?

tensorflow - 不理解 TensorFlow MNIST 指南中使用的代码

python - 在 PyTorch 中索引多维张量中的最大元素

ubuntu - 在 Ubuntu : make can't find Boost's include files 上构建 Caffe

Caffe的Python接口(interface) : Error in "import caffe"

python - 使用python层Caffe实现Bhattacharyya损失函数

logging - tensorflow log_softmax tf.nn.log(tf.nn.softmax(predict)) tf.nn.softmax_cross_entropy_with_logits

tensorflow - 如何比较 tensorflow 中的两个张量(不是按元素而是按张量)?

machine-learning - 多标签分类中的最后一层应该是什么?

tensorflow - TypeError : Expected any non-tensor type, 得到了一个张量