TensorFlow檢視輸入節點和輸出節點名稱方式
阿新 • • 發佈:2020-01-09
TensorFlow 定義輸入節點名稱input_name:
with tf.name_scope('input'): bottleneck_input = tf.placeholder_with_default( bottleneck_tensor,shape=[batch_size,bottleneck_tensor_size],name='Mul')
TensorFlow檢視pb資料庫裡面的輸入節點和輸出節點:
import tensorflow as tf import os model_dir = './tmp/' model_name = 'output_graph.pb' def create_graph(): with tf.gfile.FastGFile(os.path.join( model_dir,model_name),'rb') as f: graph_def = tf.GraphDef() graph_def.ParseFromString(f.read()) tf.import_graph_def(graph_def,name='') create_graph() tensor_name_list = [tensor.name for tensor in tf.get_default_graph().as_graph_def().node] for tensor_name in tensor_name_list: print(tensor_name,'\n')
以上這篇TensorFlow檢視輸入節點和輸出節點名稱方式就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。