我對Tensorflow非常陌生,我正在嘗試使用inception v3網絡來訓練iPhone應用程序中使用的東西。我設法我的圖形導出爲protocolbuffer文件,手動刪除差節點(正確的,我希望如此),並放置在.pb文件在我的iOS項目,但現在我收到以下錯誤:如何獲取用於iOS示例應用程序的圖層名稱? (Tensorflow)
Running model failed:Not found: FeedInputs: unable to find feed output input
這似乎表明我的input_layer_name
和output_layer_name
變量在iOS應用程序中配置錯誤。
我在各個地方看到,它應該分別爲Mul
和softmax
,對於初始版本v3,但這些值對我不起作用。
我的問題是:什麼是一層(關於這個上下文),以及如何找出我的是什麼?
This是我訓練的模型的確切定義,但我沒有看到「Mul」或「softmax」出現。
This是我已經能夠了解圖層,但它似乎是一個不同的概念,因爲「Mul」不在該列表中。
我擔心這可能是this question重複,但「層」不解釋(他們是張量?)和graph.get_operations()
似乎被棄用,或者也許我用錯了。
這是最全面的答案,並涵蓋了寶貴的觀點。我沒有使用Tensorboard,而是循環打印出圖表中的每個節點,找到名稱,最終得到的結果與您在最後的推理代碼中提供的內容類似。你的回答也突出了佔位節點的重要性,這些節點在github上的Inception v3模型中似乎不存在,所以我添加了我自己的並將其傳入。我將把你的答案作爲接受的答案。非常感謝您的幫助! –