tensorflow训练的模型在java中的使用
来源:互联网 发布:mysql delete 编辑:程序博客网 时间:2024/05/15 15:09
tensorflow训练模型通常使用python api编写,简单记录下这些模型保存后怎么在java中调用。
python中训练完成,模型保存使用如下api保存:
# 保存二进制模型output_graph_def = tf.graph_util.convert_variables_to_constants(sess, sess.graph_def, output_node_names=['y_conv_add'])with tf.gfile.FastGFile('/logs/mnist.pb', mode='wb') as f: f.write(output_graph_def.SerializeToString())保存为二进制pb文件,主要的点是output_node_names数组,该数据的名称表示需要保存的tensorflow tensor名。既是在python中定义模型时指定的计算操作的name。填写什么就保存到什么节点。在cnn模型中,通常是分类输出的名称。
例如模型定义时代码为:
y_conv = tf.add(tf.matmul(h_fc1_drop, W_fc2), b_fc2, name='y_conv_add') # cnn输出层,名称y_conv_add# 训练和评价模型softmax = tf.nn.softmax_cross_entropy_with_logits(labels=y_, logits=y_conv)
模型在java中使用需要关心模型输入tensor和输出tensor名,所以定义模型时,所有的输入tensor最好指定名称,如输入x和dropout名。
java中调用代码片段:
public static void main(String[] args) {String labels = "17,16,7,8,3,15,4,14,2,5,12,18,9,10,1,11,13,6";TensorFlowInferenceInterface tfi = new TensorFlowInferenceInterface("D:/tf_mode/output_graph.pb","imageType");final Operation operation = tfi.graphOperation("y_conv_add");Output output = operation.output(0);Shape shape = output.shape();final int numClasses = (int) shape.size(1);float[] floatValues = getImagePixel("D:/tf_mode/ci/ci/333.jpg"); //将图片处理为输入对应张量格式// 输入图片tfi.feed("x_input", floatValues, 1, 2048); //将数据复制给输入张量x_input即为模型定义时的x名称tfi.run(new String[] { "y_conv_add" }, false);//输出张量float[] outPuts = new float[numClasses];//结果分类tfi.fetch("y_conv_add", outPuts);//接收结果 outPuts保存的即为预测结果对应的概率,最大的一个通常为本次预测结果}
TensorFlowInferenceInterface参考:
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/android/java/org/tensorflow/contrib/android/TensorFlowInferenceInterface.java
java api和tensorflow的依赖:
https://github.com/tensorflow/tensorflow/tree/master/tensorflow/java
调用过程参考:
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/android/src/org/tensorflow/demo/TensorFlowImageClassifier.java
注:内容为随手记录,未检查错别字、是否遗漏等。
鉴于多人所要代码,在此提供python及java调用实现:
http://pan.baidu.com/s/1mi1hklM 交代下背景python实现为基于.pb文件的二次训练,只训练最后一层分类层的代码,项目初始为解决单个汉字的识别问题。
0 0
- tensorflow训练的模型在java中的使用
- tensorflow训练的模型在java中的使用
- TensorFlow的训练模型在Android和Java的应用及调用
- TensorFlow的训练模型在Android和Java的应用及调用
- Tensorflow实现cnn模型的训练与使用
- Tensorflow实现cnn模型的训练与使用
- tensorflow训练好的模型中java调用
- java加载tensorflow训练好的模型部署成service
- tensorflow在训练模型的时候如何指定GPU进行训练
- TensorFlow在MNIST中的应用-训练过程的可视化
- Tensorflow使用的预训练的resnet_v2_50,resnet_v2_101,resnet_v2_152等模型预测,训练
- TensorFlow在训练模型时指定GPU进行训练
- adaboost + cascade 以及在 opencv 中的使用 ------ 模型训练
- tensorflow将训练好的模型
- C++调用tensorflow 训练好的模型
- c++调用python训练的tensorflow模型
- tensorflow 模型训练
- 如何理解TensorFlow,TensorFlow在神经网络训练中的作用是什么?
- 软件性能测试的本质
- 4.0spring框架学习(四)自动装配
- App测试方法总结
- python 导入循环问题
- MYSQL数据库规范-字段规范
- tensorflow训练的模型在java中的使用
- Android NavigationView的使用详解以及各种问题解决方案
- 5.0spring框架学习(五)注解
- python中用logging实现日志滚动和过期日志删除
- springmvc发生No mapping found for HTTP request with URI[...] in dispatcher servlet
- kafka生产者客户端——那年我趟过的坑
- linux安装虚拟机及weblogic
- Physics for Game Developers 阅读
- MQTT协议-MQTT协议解析(MQTT数据包结构)