一尘不染

TensorFlow将计算图导出为XML,JSON等

json

我想将TensorFlow计算图导出为XML或类似的东西,所以我可以使用外部程序对其进行修改,然后重新导入。我找到了Meta
Graph,
但是它以二进制格式导出,我不知道如何修改。

是否存在这种能力?


阅读 321

收藏
2020-07-27

共1个答案

一尘不染

TensorFlow数据流图的本机序列化格式使用协议缓冲区,该缓冲区具有许多不同语言的绑定。您可以生成代码,该代码应能够从以下两种消息模式中解析二进制数据:(tensorflow.GraphDef一个较低级别的表示形式)和tensorflow.MetaGraphDef(一个较高级别的表示形式,其中包括GraphDef有关如何解释节点中某些节点的信息)。图形)。

如果目标语言没有协议缓冲区实现,则可以从Python协议缓冲区对象生成JSON。例如,以下代码生成一个字符串,其中包含的JSON表示形式GraphDef

import tensorflow as tf
from google.protobuf import json_format

with tf.Graph().as_default() as graph:
  # Add nodes to the graph...

graph_def = graph.as_graph_def()

json_string = json_format.MessageToJson(graph_def)
2020-07-27