部署網絡缺少一個Softmax層

我嘗試使用pretrained模型（VGG 19）來DIGITS，但我得到了這個錯誤。部署網絡缺少一個Softmax層

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

我嘗試使用只有兩個類的數據集進行測試。

我讀this和this嘗試修改最後一層，但也得到錯誤。我如何根據新的數據集來修改圖層？

我嘗試修改的最後一層，我得到了錯誤

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.

來源

2016-10-20 Video Analysis Deep Learning

你有，因爲你想上傳一個「火車/ VAL」網絡出現了問題，當你確實需要上傳「一體化」網絡。不幸的是，我們沒有很好地記錄這一點。我創建了an RFE以提醒我們改進文檔。

嘗試調整最後一層網絡中看起來是這樣的：https://github.com/NVIDIA/DIGITS/blob/v4.0.0/digits/standard-networks/caffe/lenet.prototxt#L162-L184

欲瞭解更多信息，here是我是如何提出的更新來自Caffe的例子網絡的所有功能於一身的漁網，並here是我如何將默認的DIGITS網絡更新爲一體化網絡。

來源

2016-10-20 01:30:19

謝謝，但我得到錯誤，當我改變最後一層。「錯誤：圖層'softmax'在TRAIN階段引用底部'fc8'，但該階段不包含此blob，請考慮使用include指令來限制此圖層的範圍。我怎樣才能改變最後一層？ –

如何更改最後一層？我怎樣才能使用「一體化」？與微調模型相關的「一體式」？對於使用VGG 19層，我如何修改最後一層以便在僅有兩個類的數據集中使用？ –

你可能會發現[這個答案]（http://stackoverflow.com/a/33773152/1714410）很有用：如何將「train」原型文件轉換爲「部署」原型文件。 – Shai

部署網絡缺少一個Softmax層

回答

相關問題