Converting trained Tensorflow model to protobuf

Question

The inception net in the Android Tensorflow Demo stores the model as a protobuf file (tensorflow_inception_graph.pb). I'd like to replace this network with another one.

Is there documentation or an example somewhere on how to convert a trained TensorFlow network in Python to .pb? I'm aware of TensorFlow's Saver but this seems to be used to save intermediate training state. Not sure how it works if the model is already trained.

if one of the mentioned answers is correct or you found a different solution and update here would be nice — BraveHeart, Apr 07 '16 at 19:54

score 5 · Answer 1 · answered Feb 07 '16 at 00:16

5

Here's an example of saving and loading

answered Feb 07 '16 at 00:16

Yaroslav Bulatov

57,332
22
139
197

score 4 · Accepted Answer · answered Feb 20 '16 at 15:14

4

tf.Graph.as_graph_def() retrieves the serialized graph. Then you just write it to a file.

answered Feb 20 '16 at 15:14

Jay Alammar

81
4

Converting trained Tensorflow model to protobuf

2 Answers2

Linked