android - Caffe:检查失败:网络应该只有一个输出

标签 android machine-learning neural-network deep-learning caffe

我正在使用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/

相关文章:

android - Android 中未发送调用

Android ViewFlipper动画(ViewPage)

android Wifilock-移动数据访问锁怎么样?

python - 使用预训练模型在 tensorflow 中训练新模型

python - 医院等候时间的情绪分析

android - 在 Android Studio 中单击 ListFragment 中的项目时显示详细信息 fragment

machine-learning - Isomap 中的维数

neural-network - 带有多标签图像的咖啡

python - 使用 Keras : All layer names should be unique for discriminator 在 GPU 上训练 GAN

python - pyTorch 中的矩阵乘法