我正在使用Caffe android lib (基于 C++ API)。我正在尝试使用 GoogleNet 模型。
不幸的是,它给了我这个错误:
Check failed: net_->num_outputs() == 1 Network should have exactly one output
我感觉我错过了一些愚蠢的东西。有什么想法吗?
最佳答案
您使用的是哪种 GoogLeNet 实现?
根据实现的不同,您可能确实会得到多个输出。
原始的 GoogLeNet(在 CVPR 2015 中发布)具有三个用于训练的“输出”损失,因此您可能会使用具有多个输出层的类似原型(prototype)文本。
考虑this deploy.prototxt
- 它只有一个输出“prob”
。然而,对应的train_val.prototxt
有三个输出:“loss1/loss1”
、“loss2/loss1”
和 “loss3/loss3”
。
关于android - Caffe:检查失败:网络应该只有一个输出,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/34637573/