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

Question

我想重新组织tensorflow .pb模型的节点,所以我首先从GraphDef获取NodeDef,然后使用NodeDef.attr()来获取"Conv2D"的节点.我能得到的参数,如进展,填充,DATA_FORMAT,use_cudnn_on_gpu从ATTR,但不能获取的权重格式参数.我使用的语言是c ++.怎么弄它!谢谢!

Answer 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()