Wie erstelle ich classify_image_graph_def.pb aus meinem Modell?

Question

Ich habe this repo verwendet, um mein Caffe-Modell in Tensorflow umzuwandeln. Ich endete mit 2 Dateien: eine ist Python-Klasse und die zweite ist Npy-Datei mit den Modellgewichten.

Allerdings möchte ich eine einzige Datei mit dem gleichen Format wie diese one generieren (diese Datei, genannt classify_image_graph_def.pb, kann verwendet werden, um das Netz über ein beliebiges Testbild weiterzuleiten).

Ich interessiere mich für dieses Format, weil dies das Skript quantize_graph.py erforderlich ist.

Answer 1

Der richtige Weg, um es zu tun ist, alle Variablen in der Grafik in Konstante zu konvertieren. Dies könnte unter Verwendung der freeze_graph tool erfolgen. Da aber dieses Tool ist nicht Teil der Laufzeit tensorflow Bibliothek o es bequemer fand, dass es mit den folgenden Zeilen zu tun (ohne freeze_graph bauen zu müssen):

sess = tf.InteractiveSession() 

### create some graph here ### 
############################## 

graph_def = sess.graph.as_graph_def() 
output_node_names = "output0,output1" # put the names of the output nodes here 

# freeze all parameters and save 
output_graph_def = graph_util.convert_variables_to_constants(
     sess, graph_def, output_node_names.split(",")) 
with tf.gfile.GFile(output_graph_file, "wb") as f: 
    f.write(output_graph_def.SerializeToString())