我搜索了一个星期,以从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
我没有占位符输出节点。