export frozen graph

Author: memo

pix2pix.py

@@ -644,6 +644,10 @@ def main():
             export_saver.export_meta_graph(filename=os.path.join(a.output_dir, "export.meta"))
             export_saver.save(sess, os.path.join(a.output_dir, "export"), write_meta_graph=False)
 
+            # write frozen graph
+            graph_frz = tf.graph_util.convert_variables_to_constants(sess, sess.graph_def, ['generator/generator_outputs'])
+            tf.train.write_graph(graph_frz, a.output_dir, 'graph_frz.pb', as_text=False)
+
         return
 
     examples = load_examples()