Wie erhält man das Gewichtungsformat vom TensorFlow .pb-Modell?

Question

Ich möchte die Knoten von Tensorflow. Pb-Modell reorganisieren, so dass ich zuerst NodeDef von GraphDef, und erhalten Sie attribution NodeDef.attr(). Für den Knoten von "Conv2D". Ich kann Parameter wie strides, padding, data_format, use_cudnn_on_gpu von attr, aber nicht erhalten die Gewichte Formatparameter. Die Sprache, die ich verwende, ist C++. Wie man es bekommt! Vielen Dank!

Answer 1

Conv2D hat zwei Eingänge: der erste ist, Daten und die zweite ist filter (oder Gewichte), so kann man einfach das Format von dem zweiten Eingang des Conv2D überprüfen. Wenn Sie C++ verwenden, können Sie Folgendes versuchen:

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