save_graph.py

# function ffwd, line 98
# https://github.com/lengstrom/fast-style-transfer/blob/master/evaluate.py#L98
if os.path.isdir(checkpoint_dir):
    ckpt = tf.train.get_checkpoint_state(checkpoint_dir)
    if ckpt and ckpt.model_checkpoint_path:
        saver.restore(sess, ckpt.model_checkpoint_path)
        ########## add this for pre-trained models ###########
        frozen_graph_def = tf.graph_util.convert_variables_to_constants(sess,sess.graph_def,['add_37'])
        with open('output_graph.pb', 'wb') as f:
            f.write(frozen_graph_def.SerializeToString())
        #####################################################
  else:
    raise Exception("No checkpoint found...")
else:
       saver.restore(sess, checkpoint_dir)
       ########## add this for custom models ###########
       frozen_graph_def = tf.graph_util.convert_variables_to_constants(sess,sess.graph_def,['add_37'])
       with open('output_graph.pb', 'wb') as f:
           f.write(frozen_graph_def.SerializeToString())
       #####################################################