4.python tensorflow 查看每一個節點

# @File  : node.py
# @Author: Wang Zhimin
# @Date  : 2019/8/28
from tensorflow.python import pywrap_tensorflow
import os
checkpoint_path=os.path.join(r'C:\Users\wangz\Desktop\YOLOv3_TensorFlow-master\data\darknet_weights\cache.ckpt-2000')
reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map=reader.get_variable_to_shape_map()
for key in var_to_shape_map:
    print('tensor_name: ',key)