2 回答

TA貢獻(xiàn)1827條經(jīng)驗(yàn) 獲得超8個(gè)贊
該錯(cuò)誤表明Output
(某個(gè)節(jié)點(diǎn)的)屬性是 nil 操作。
因此graphModel.Operation("input").Operation(0)
或graphModel.Operation("output").Output(0)
返回nil
。
要更正此問(wèn)題,您必須引用圖中的現(xiàn)有節(jié)點(diǎn),因?yàn)閳D中沒(méi)有命名的張量input
或命名的張量。output
從您用于導(dǎo)出模型的 Python 代碼中,您可以找到輸入和輸出張量的完整名稱。只需訪問(wèn).name
輸入占位符和輸出節(jié)點(diǎn)的屬性,即可獲得在 Go 中使用的正確名稱。
此外,Go 綁定使用起來(lái)很復(fù)雜,尤其是當(dāng)您想對(duì)輸入圖像運(yùn)行一些預(yù)處理操作時(shí)。

TA貢獻(xiàn)1785條經(jīng)驗(yàn) 獲得超4個(gè)贊
需要這樣做:my_model.inputs
并my_model.outputs
獲得正確的名稱。例如:
>?my_model.inputs [<tf.Tensor?'dense_1_input:0'?shape=(?,?7)?dtype=float32> >?my_model.outputs [<tf.Tensor?'my_output/BiasAdd:0'?shape=(?,?2)?dtype=float32>
因此,我的輸入和輸出節(jié)點(diǎn)是dense_1_input
和my_output/BiasAdd
(不是my_output
?。?/p>
- 2 回答
- 0 關(guān)注
- 209 瀏覽
添加回答
舉報(bào)