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 关注
- 2124 浏览
添加回答
举报
0/150
提交
取消