Caffe：檢查失敗：網絡應該只有一個輸出

我使用的是Caffe android lib（基於C++ api）。我正在用GoogleNet模型嘗試它。
不幸的是它給我這個錯誤：Caffe：檢查失敗：網絡應該只有一個輸出

Check failed: net_->num_outputs() == 1 Network should have exactly one output

我有我失去了一些愚蠢的感覺。有什麼想法嗎？

來源

2016-01-06 Mozie

您使用的是什麼GoogLeNet的實現？
根據實施情況，您可能確實獲得的不止是單個輸出。最初的GoogLeNet（發佈於CVPR 2015）有三個「輸出」損失用於培訓，因此您可能會使用由多個輸出層產生的similat原型文件。

考慮到this deploy.prototxt - 它只有一個輸出"prob"。但是，相應的train_val.prototxt有三個輸出："loss1/loss1","loss2/loss1"和"loss3/loss3"。

來源

2016-01-06 16:24:14 Shai

感謝您的快速回復，我確實切換回部署，但它仍然給我那個錯誤。 Wierd的一部分，該網絡在該圖書館的例子中使用：https://github.com/BVLC/caffe/blob/master/models/bvlc_reference_caffenet/deploy.prototxt不是 – Mozie

@Mozie你應該檢查你的'deploy.prototxt' - 它可能不止一個輸出，這並不罕見。 – Shai

謝謝Shai，最後一層確實給了多個，謝謝！（知道我是在愚蠢的地方） – Mozie

Caffe：檢查失敗：網絡應該只有一個輸出

回答

相關問題