tensorflow打印所有变量名_TensorFlow(一)模型参数名称和数值读取

1. 模型参数名称读取(新式4个模型文件)

image.png

import os

import re

import tensorflow as tf

from tensorflow.python import pywrap_tensorflow

model_exp = "./log/vgg16/fine_tune"

model_exp = "model_vgg16"

#model_exp = "model-lenet"

def get_model_filenames(model_dir):

files = os.listdir(model_dir)

meta_files = [s for s in files if s.endswith('.meta')]

if len(meta_files) == 0:

#raise load_modelValueError('No meta file found in the model directory (%s)' % model_dir)

print("No meta file found in the model directory ")

elif len(meta_files) > 1:

raise ValueError('There should not be more than one meta file in the model directory (%s)' % model_dir)

meta_file = meta_files[0]

ckpt =