Trainiertes Tensorflow-Modell in Protobuf umwandeln

8

Das Anfangsnetz in der Android Tensorflow-Demo speichert das Modell als eine Protobuf-Datei ( tensorflow_inception_graph.pb ). Ich möchte dieses Netzwerk durch ein anderes ersetzen.

Gibt es Dokumentation oder ein Beispiel, wie man ein trainiertes TensorFlow-Netzwerk in Python in .pb umwandelt? Ich kenne TensorFlows Saver , aber das scheint daran gewöhnt zu sein Zwischentrainingszustand sparen. Nicht sicher, wie es funktioniert, wenn das Modell bereits trainiert ist.

    
Zach Rattner 06.02.2016, 22:41
quelle

2 Antworten

4

tf.Graph.as_graph_def() ruft das serialisierte Diagramm ab. Dann schreibst du es einfach in eine Datei.

    
Jay Ammar 20.02.2016, 15:14
quelle
5

Hier ist ein Beispiel für Speichern und loading

    
Yaroslav Bulatov 07.02.2016 00:16
quelle

Tags und Links