TensorFlow Export Compute-Grafik in XML, JSON, etc

Question

Ich möchte ein TensorFlow-Compute-Diagramm in XML oder etwas ähnliches exportieren, damit ich es mit einem externen Programm ändern und dann neu importieren kann. Ich fand Meta Graph, aber das exportiert in einem binären Format, das ich nicht würde ändern können.

Gibt es solche Möglichkeiten?

Answer 1

Das native Serialisierungsformat für das TensorFlow-Datenflussdiagramm verwendet protocol buffers, die Bindungen in vielen verschiedenen Sprachen enthalten. Sie können Code generieren, der in der Lage sein soll, die binären Daten aus den zwei Meldungsschemas zu analysieren: tensorflow.GraphDef (eine untergeordnete Darstellung) und tensorflow.MetaGraphDef (eine übergeordnete Darstellung, die eine GraphDef und andere Informationen zur Interpretation einiger der enthält Knoten in der Grafik).

Wenn für Ihre Zielsprache keine Protokollpufferimplementierung vorhanden ist, können Sie JSON aus dem Python-Protokollpufferobjekt generieren. Zum Beispiel erzeugt die folgende eine Zeichenfolge eine JSON-Darstellung eines GraphDef enthält:

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)