天天看點

tf 檢視模型内容

from tensorflow.python.tools.inspect_checkpoint import print_tensors_in_checkpoint_file

savedir = '/Users/qyk/Desktop/py/models/inception_v3/'

print_tensors_in_checkpoint_file(savedir + 'inception_v3.ckpt',tensor_name=None,all_tensors=True)

####

tensor_name: InceptionV3/AuxLogits/Conv2d_1b_1x1/BatchNorm/beta

[.....]

tensor_name: InceptionV3/AuxLogits/Conv2d_1b_1x1/BatchNorm/moving_mean

tensor_name: InceptionV3/AuxLogits/Conv2d_1b_1x1/BatchNorm/moving_variance

.....

tensor_name: InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/BatchNorm/moving_variance

tensor_name: InceptionV3/Mixed_7c/Branch_3/Conv2d_0b_1x1/weights

tensor_name: global_step