tensorflow权重文件的解读

1.解读tensorflow权重文件,透过 tf.train.NewCheckpointReader函数。
2.reader.get_variable_to_shape_map()可以得到权重文件里面的tensor名称。
3.reader.get_tensor(key) 可以得到对应tensor的权重值。

import tensorflow as tf

cpktFileName = r'.\models\resnet_v2_152.ckpt'
reader = tf.train.NewCheckpointReader(cpktFileName)
for key in sorted(reader.get_variable_to_shape_map()):
   
    if key.endswith('weights') or key.endswith('biases'):
        keySplits = key.split(r'/')
        print(key)
        print(reader.get_tensor(key))
        

4. resnet_v2_152权重 tensor name解读
   第一，每个tensor name都以resnet_v2_152开头
   第二，tensor name第二段为block，共有四个block。与网络架构有关。
   第三，第三字段为unit，每个block里面unit数量不同。与网络架构有关。
   第四，除了组后的平坦层，第四字段都为bottleneck_v2
   第五，第五字段为‘conv1’，‘conv2’，‘conv3’，‘shortcut’
   第六，第六字段为‘weights’ or ‘biases’