全部!这些天我遇到了问题。 问题是,使用Tensorflow加速基于ARM的目标的推理。 我正在练习 tfcompile 从this dude's page引用,我使用:
tfcompile --graph=test_graph.pb --config=test_graph.pbtxt --entry_point=test_func --cpp_class=test --out_object=test_func.o --out_header=test.hpp
test_graph.pbtxt是由以下人员生成的:
user@user $ python
>>>
>>> import tensorflow as tf
>>> from tensorflow.python.platform import gfile
>>> from google.protobuf import text_format
>>> filename='/home/user/test_graph.pb'
>>> with gfile.FastGFile(filename,'rb') as f:
... graph_def = tf.GraphDef()
... graph_def.ParseFromString(f.read())
... tf.import_graph_def(graph_def, name='')
... tf.train.write_graph(graph_def, '/home/user/', 'test_graph.pbtxt', as_text=True)
>>>
当我运行tfcompile
时
我得到了错误:
Non-OK-status: status status: Data loss: Can't parse /home/user/test_graph.pbtxt as text proto
因此* .pbtxt不应从* .pb转换。
我搜索--config=
应该是什么,但没有结果。
你能帮我解决吗?
提前谢谢!