java 操作 TensorFlow 合適嗎
1 回答

holdtom
TA貢獻1805條經驗 獲得超10個贊
tensorflow訓練模型通常使用Python api編寫,這些模型保存后可以在Java中調用。
python中訓練完成,模型保存使用如下api保存:
保存為二進制pb文件,主要的點是output_node_names數組,該數據的名稱表示需要保存的tensorflow tensor名。既是在python中定義模型時指定的計算操作的name。填寫什么就保存到什么節點。在cnn模型中,通常是分類輸出的名稱。
例如模型定義時代碼為:
模型在java中使用需要關心模型輸入tensor和輸出tensor名,所以定義模型時,所有的輸入tensor最好指定名稱,如輸入x和dropout名。
java中調用代碼片段:
- 1 回答
- 0 關注
- 2238 瀏覽
添加回答
舉報
0/150
提交
取消