TensorFlow .pb modelinden ağırlık biçimi nasıl alınır?

Question

Tensorflow .pb model düğümlerini yeniden düzenlemek istiyorum, bu yüzden ilk önce GraphDef'den NodeDef alıyorum ve "Conv2D" düğümü için NodeDef.attr() öğesini kullanın. Adımlar, dolgu, data_format, use_cudnn_on_gpu gibi parametreler attr'dan alınabilir, ancak ağırlık biçimi parametrelerini alamıyorum. Kullandığım dil C++. Nasıl edinilir? Teşekkür ederim!

Answer 1

Conv2D iki girişe sahiptir: birinci veri ve ikincisi filter (veya ağırlıklar), bu nedenle Conv2D ikinci girişinin biçimini kontrol edebilirsiniz. C++ kullanıyorsanız, bunu deneyebilirsiniz:

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