TensorFlow .pb मॉडल से वजन प्रारूप कैसे प्राप्त करें?

Question

मैं tensorflow .pb मॉडल के नोड्स को पुनर्गठित करना चाहता हूं, इसलिए मुझे पहले ग्राफडिफ से नोडडिफ मिलता है, और "Conv2D" के नोड के लिए attr NodeDef.attr() का उपयोग करें। मैं एटीआर से स्ट्रिड्स, पैडिंग, डेटा_फॉर्मेट, use_cudnn_on_gpu जैसे पैरामीटर प्राप्त कर सकता हूं, लेकिन वजन प्रारूप पैरामीटर नहीं प्राप्त कर सकता। मैं जिस भाषा का उपयोग करता हूं वह सी ++ है। इसे कैसे प्राप्त करें! धन्यवाद!

Answer 1

Conv2D में दो इनपुट हैं: पहला डेटा है और दूसरा filter (या वजन) है, ताकि आप Conv2D के दूसरे इनपुट के प्रारूप को आसानी से देख सकें। यदि आप सी ++ का उपयोग कर रहे हैं, तो आप इसे आजमा सकते हैं:

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