TensorFlow获取加载模型中的全部张量名称代码
核心代码如下：
[tensor.namefortensorintf.get_default_graph().as_graph_def().node]
实例代码：（加载了Inceptino_v3的模型，并获取该模型所有节点的名称）
#-*-coding:utf-8-*- importtensorflowastf importos model_dir='C:/Inception_v3' model_name='output_graph.pb' #读取并创建一个图graph来存放训练好的Inception_v3模型（函数） defcreate_graph(): withtf.gfile.FastGFile(os.path.join( model_dir,model_name),'rb')asf: #使用tf.GraphDef()定义一个空的Graph graph_def=tf.GraphDef() graph_def.ParseFromString(f.read()) #Importsthegraphfromgraph_defintothecurrentdefaultGraph. tf.import_graph_def(graph_def,name='') #创建graph create_graph() tensor_name_list=[tensor.namefortensorintf.get_default_graph().as_graph_def().node] fortensor_nameintensor_name_list: print(tensor_name,'\n')
输出结果：
mixed_8/tower/conv_1/batchnorm/moving_variance mixed_8/tower/conv_1/batchnorm r_1/mixed/conv_1/batchnorm . . . mixed_10/tower_1/mixed/conv_1/CheckNumerics mixed_10/tower_1/mixed/conv_1/control_dependency mixed_10/tower_1/mixed/conv_1 pool_3 pool_3/_reshape/shape pool_3/_reshape input/BottleneckInputPlaceholder . . . . final_training_ops/weights/final_weights final_training_ops/weights/final_weights/read final_training_ops/biases/final_biases final_training_ops/biases/final_biases/read final_training_ops/Wx_plus_b/MatMul final_training_ops/Wx_plus_b/add final_result
由于结果太长了，就省略了一些。
如果不想这样print输出也可以将其写入txt查看。
写入txt代码如下：
tensor_name_list=[tensor.namefortensorintf.get_default_graph().as_graph_def().node] txt_path='./txt/节点名称' full_path=txt_path+'.txt' fortensor_nameintensor_name_list: name=tensor_name+'\n' file=open(full_path,'a+') file.write(name) file.close()
以上这篇TensorFlow获取加载模型中的全部张量名称代码就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。