TensorFlow - 导出

  • 简述

    在这里,我们将重点介绍 TensorFlow 中的 MetaGraph 形成。这将帮助我们理解 TensorFlow 中的导出模块。元图包含基本信息,这些信息是在先前训练的图上训练、执行评估或运行推理所需的。
    以下是相同的代码片段 -
    
    def export_meta_graph(filename = None, collection_list = None, as_text = False): 
       """this code writes `MetaGraphDef` to save_path/filename. 
       
       Arguments: 
       filename: Optional meta_graph filename including the path. collection_list: 
          List of string keys to collect. as_text: If `True`, 
          writes the meta_graph as an ASCII proto. 
       
       Returns: 
       A `MetaGraphDef` proto. """
    
    下面提到了相同的典型使用模型之一 -
    
    # Build the model ... 
    with tf.Session() as sess: 
       # Use the model ... 
    # Export the model to /tmp/my-model.meta. 
    meta_graph_def = tf.train.export_meta_graph(filename = '/tmp/my-model.meta')