|
import os |
|
import tensorflow as tf |
|
|
|
export_dir = 'export_dir' |
|
trained_checkpoint_prefix = '/m/wk/1/model_gia-young-picasso-v03-201216-var2_new_240000.ckpt-240000' |
|
|
|
|
|
graph = tf.Graph() |
|
loader = tf.train.import_meta_graph(trained_checkpoint_prefix + ".meta" ) |
|
sess = tf.Session() |
|
loader.restore(sess,trained_checkpoint_prefix) |
|
builder = tf.saved_model.builder.SavedModelBuilder(export_dir) |
|
builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING], strip_default_attrs=True) |
|
builder.save() |