如何从 TensorFlow .pb 模型中获取权重格式？

【问题标题】：How to get weights format from TensorFlow .pb model?如何从 TensorFlow .pb 模型中获取权重格式？
【发布时间】：2017-07-12 07:52:07
【问题描述】：

我想重新组织tensorflow .pb模型的节点，所以我先从GraphDef中获取NodeDef，并获取attr use NodeDef.attr().for "Conv2D"的节点。我可以从 attr 获取 strides、padding、data_format、use_cudnn_on_gpu 等参数，但无法获取权重格式参数。我使用的语言是 c++。如何得到它！谢谢！

【问题讨论】：

标签： c++ tensorflow model

【解决方案1】：

Conv2D 有两个输入：第一个是数据，第二个是filter（或权重），因此您可以简单地检查Conv2D 的第二个输入的格式。如果你使用的是 C++，你可以试试这个：

# Assuming inputs: conv2d_node, node_map.
filter_node_name = conv2d_node.input(1)
filter_node = node_map[filter_node_name]
# You might need to check identity node here.
# Get the shape of filter_node using NodeDef.attr()

【讨论】：