将Tensorflow冻结图转换为Tensorflow JS模型

时间:2018-12-17 14:03:05

标签: tensorflow

我搜索了一个星期,以从tensorflow冻结图或检查点创建tfjs导出。我使用了更快的rcnn-Inception v2来为我自己的对象转移object_detection上的知识。

我的冻结图工作正常。我无法做的是通过tensorflow_converter传输图形。我在这里需要帮助。我确实打印了所有输出节点,但没有得到我需要在这里使用的输出:

tensorflowjs_converter --input_format=tf_saved_model --output_node_names=HERE --saved_model_tags=serve frozen_inference_graph.pb ./web_model

我还试图使以下工作正常进行,但是在这里,我又需要使用哪个节点?也许我只是完全在线,看不到它。所有帮助将不胜感激。

 print(tf.contrib.graph_editor.get_tensors(tf.get_default_graph())) 
 print(sess.graph.get_operations())

 then optimize the graph and reduce complexity
 python -m tensorflow.python.tools.optimize_for_inference --input=model.pb --output=optimized_graph.pb --input_names="Placeholder" --output_names="loss"

 finally export to web format
 tensorflowjs_converter --input_format=tf_frozen_model --output_node_names='loss' --saved_model_tags=serve ./optimized_graph.pb ./js

我没有占位符输出节点。

0 个答案:

没有答案