How to convert .ckpt to .pb?

前端 未结 3 914
渐次进展
渐次进展 2020-12-30 13:24

I am new to deep learning and I want to use a pretrained (EAST) model to serve from the AI Platform Serving, I have these files made available by the developer:

3条回答
  •  孤城傲影
    2020-12-30 14:08

    Following the answer of @Puneith Kaul, here is the syntax for tensorflow version 1.7:

    import os
    import tensorflow as tf
    
    export_dir = 'export_dir' 
    trained_checkpoint_prefix = 'models/model.ckpt'
    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.TRAINING, tf.saved_model.tag_constants.SERVING], strip_default_attrs=True)
    builder.save()
    

提交回复
热议问题