分类：权重文件

2021
07-17

基于tensorflow权重文件的解读

1.解读tensorflow权重文件,透过tf.train.NewCheckpointReader函数。2.reader.get_variable_to_shape_map()可以得到权重文件里面的tensor名称。3.reader.get_tensor(key)可以得到对应tensor的权重值。importtensorflowastfcpktFileName=r'.\models\resnet_v2_152.ckpt'reader=tf.train.NewCheckpointReader(cpktFileName)forkeyinsorted(reader.get_variable_to_shape_map()):ifkey.endswith('weights')orkey... 继续阅读 >