我的项目使用 Python 在 TensorFlow 上训练 MLP，然后我以这种方式导出图表和权重:
tf.train.write_graph(sess.graph_def, "./", "inp.txt", True)
saver.save(sess, 'variables/model.ckpt', global_step=1)
现在，虽然可以使用这两个文件将其导入回 Python，但似乎无法将其用于 Android 或 C++，因为它无法导入检查点 .ckpt。
现在，我正在使用 google 提供的脚本 freeze_graph.py 通过执行以下操作将两个文件合二为一:
bazel-bin/tensorflow/python/tools/freeze_graph --input_graph=inp.txt --input_checkpoint=variables/model.ckpt-1 --output_graph=newoutput.pb --output_node_names=output
我的问题是，有没有办法使用另一个函数而不是 tf.train.write_graph 来导出它并包含权重？
最佳答案
关于android - 为 TensorFlow 导出的 .pb 文件添加权重，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/40409078/