部署网络缺少 Softmax 层

Deploy network is missing a Softmax layer

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

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

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

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

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

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 is how I've proposed updating Caffe's example networks to all-in-one nets, and here 是我将默认 DIGITS 网络更新为一体化网络的方式。