java调用tensorflow模型jdk必须是1.8吗? - 爱问答

(爱问答)

java调用tensorflow模型jdk必须是1.8吗?

tensorflow训练模型通常使用Python api编写,这些模型保存后可以在java中调用。python中训练完成,模型保存使用如下api保存:1234# 保存二进制模型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模型中,通常是分类输出的名称。例如模型定义时代码为:123y_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中调用代码片段:12345678910111213public 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保存的即为预测结果对应的概率,最大的一个通常为本次预测结果


相关标签:java

下一篇:H3Cs3848g出现一下情况是什么情况敲回车没反应

上一篇:dynamo能自动给构件属性赋值吗?

热门标签:
excel 网盘 破解 word dll
最新更新:
微软重新评估新的Outlook的使用时机 联想推出搭载联发科Helio G80芯片组的Tab M9平板 英特尔创新大赛时间确定! 微软Edge浏览器在稳定渠道中推出Workspaces功能 英伟达RTX4060TiGPU推出MaxSun动漫主题! 谷歌地图为用户提供了街景服务! GameSir 在T4 Kaleid中推出了一款出色的控制器! 微软开始在Windows 11 中测试其画图应用程序的新深色模式! LG电子推出全球首款无线OLED电视 英伟达人工智能芯片崭露头角! Steam Deck可以玩什么游戏-Steam Deck价格限时优惠 雷蛇推出CobraPro鼠标 Kindle电子阅读器可以访问谷歌商店吗 Windows10如何加入组策略 window10图片查看器怎么没有了?