tensorflow - 像python中的xml etree一样解析pbtxt(tensorflow)

Question

我正在使用 pbtxt 格式的 TensorFlow 图。

pbtxt 文件是从 tf 对象检测 API 生成的，我需要像 xml etree 一样解析它。我怎样才能做到这一点？

.pbtxt 是 freeze_graph 的结果

Answer 1

问题解决了。您可以使用:

import tensorflow as tf
from google.protobuf import text_format

with open('graph.pbtxt') as f:
    txt = f.read()

graph_def = text_format.Parse(txt, tf.GraphDef())

for node in graph_def.node:
    if node.name == "bla-bla":
        #TODO

更多信息: here