使用从java+(spring)嵌入的tensorflow库执行python程序不起作用

Question

>

首先，我在spring框架中有一个java程序，它试图调用一个简单的hello world python脚本。

它工作了，并返回一个“Hello World”

所以我尝试运行一个嵌入了tensorflow库的python脚本，但它没有返回任何内容（控制台中没有打印任何内容）

Python+TensorFlow

将tensorflow导入为tf导入sys导入os

OS.Environ['TF_CPP_MIN_LOG_LEVEL']=“2”将tensorflow导入为tf

label_lines=[line.rstrip（）用于tf.gfile.gfile（“logs/trained_labels.txt”）中的行]

使用tf.gfile.fastgfile（“logs/trained_graph.pb”,'rb')作为f:graph_def=tf.graphdef()graph_def.parsefromstring(f.read())_=tf.import_graph_def(graph_def,name='')

使用tf.session（）作为SESS：

softmax_tensor = sess.graph.get_tensor_by_name('final_result:0')

predictions = sess.run(softmax_tensor, \
         {'DecodeJpeg/contents:0': image_data})

top_k = predictions[0].argsort()[-len(predictions[0]):][::-1]

for node_id in top_k:
    human_string = label_lines[node_id]
    score = predictions[0][node_id]
    print('%s (Percentage = %.2f )'  %  (human_string, (score * 100 )))

java脚本

软件包COM.TEST；导入java.io.IOException；导入java.io.*；

public static void main(String [] args) throws InterruptedException {

        BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));

问题-我能运行一个从java程序嵌入了tensorflow的python吗？

Answer 1

您可以加载保存的tensorflow模型，并从Java运行预测。检查这里

加载train_graph.pb文件，并使用feed和fetch函数发送和接收数据。您可以按照android教程操作