android - 在使用 tensorflow 模型时，我们可以使用 .pbtxt 而不是 .pb 文件吗

Question

我创建了一个生成 .pbtxt 文件的 Tensorflow 模型。我可以在构建 Android 应用程序时使用此文件，通过将其重命名为 .pb 文件来使用生成的模型吗？

提前致谢。

Answer 1

如果 .pbtxt 文件确实是 TensorFlow 图的文本表示，那么不，Android API 目前不接受它，而是需要图的二进制表示。

也就是说，如果您有 .pbtxt 文件，您可以使用几行 Python 轻松地将其转换为二进制 Protocol Buffer :

import tensorflow as tf
from google.protobuf import text_format

with open('/tmp/myfile.pbtxt') as f:
  txt = f.read()
gdef = text_format.Parse(txt, tf.GraphDef())

tf.train.write_graph(gdef, '/tmp', 'myfile.pb', as_text=False)

或者，如果您控制生成 pbtxt 文件的管道，也许您可​​以更改它以以二进制格式写出文件？

希望对您有所帮助。