c++ - 使用不同 deploy.prototxt 文件的 Caffe C++ 示例出错

Question

我trained使用 MNIST 的模型 example architecture (但在我自己的一组 3 个图像类上)并一直在尝试将其集成到 C++ example 中.我修改了 MNIST 架构文件，使其类似于 C++ example 的 deploy.prototxt 文件(replacing the train and test layers with the input layer)。

不幸的是，当我运行 C++ 程序时，出现以下错误:

F0827 14:57:28.427697 25511 insert_splits.cpp:35] Unknown bottom blob 'label' (layer 'accuracy', bottom index 1)

我尝试用谷歌搜索它，我认为 MNIST 和 C++ 示例文件中的层之间存在一些差异，但无法弄清楚我可以更改哪些内容来实现这项工作。

Answer 1

正如 AbdulRahman AlHamali's comment 指出的那样看起来你在 deploy.prototxt 文件中留下了期望作为输入的损失层和准确度层(“bottom”s)“label”。
从 deploy.prototxt 中删除这些损失层应该可以解决问题。

请注意，如果您使用 "SoftmaxWithLoss" 层作为损失，您需要将其替换为 "Softmax" 层以获得类别概率作为净输出。 "Softmax" 层只需要一个 "bottom" 并且不需要 bottom: "label"。