TensorFlow .pbモデルからウェイトフォーマットを取得するには？

Question

テンソルフローの.pbモデルのノードを再構成したいので、まずGraphDefからNodeDefを取得し、attrが "Conv2D"のノードに対してNodeDef.attr（）を使用するようにします。 私はstrides、padding、data_format、use_cudnn_on_gpuなどのパラメータをattrから取得できますが、ウェイトフォーマットパラメータは取得できません。 私が使用する言語はC++です。 入手方法ありがとうございました！

Answer 1

Conv2Dには2つの入力があります。最初のデータはfilter（または重み）です。したがって、2番目の入力の形式は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()