Tensorflow 1.0之后模型文件、权重数值的读取方式
保存的文件有4个：
checkpoint
model-parameters.bin-46000.data-00000-of-00001
model-parameters.bin-46000.index
model-parameters.bin-46000.meta
读取代码：
fromtensorflow.pythonimportpywrap_tensorflow
withtf.Graph().as_default():
withtf.Session()assess:
ckpt=tf.train.get_checkpoint_state('./checkpoint')
ifckptandckpt.model_checkpoint_path:
reader=pywrap_tensorflow.NewCheckpointReader('./checkpoint/model-parameters.bin-46000')
all_variables=reader.get_variable_to_shape_map()
w1=reader.get_tensor("conv1/weight")
print(w1.shape)
print(w1)
print(all_variables)
else:
print('Nocheckpointfilefound')
以上这篇Tensorflow1.0之后模型文件、权重数值的读取方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。