Export graph

[hvico]

Hello,

I'm trying to export this model and its weights into a frozen graph. So far I did this in the "save images" part of the training loop:

saver = tf.train.Saver()
saver.save(sess, 'my_stn_model_' + str(epoch))
tf.train.write_graph(sess.graph_def, ".", "test.pb", False) #proto

Then I have my pb and the weights. But I am unable to generate a frozen graph from this because I cannot guess the output node in the graph (which is a really complex one it seems).

I attached the pb which I am able to visualize using Netron:

test.zip

Thanks in advance.