android – 将权重添加到TensorFlow导出的.pb文件中

前端之家收集整理的这篇文章主要介绍了android – 将权重添加到TensorFlow导出的.pb文件中前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。
我的项目使用 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来导出它包含的权重?

解决方法

目前我很抱歉告诉你没办法.

如前所述,在Github(have a look)上,TensorFlow团队目前没有解决这个问题.

现在不能说什么

pltrdy

原文链接:https://www.f2er.com/android/311565.html

猜你在找的Android相关文章