查看tensorflow 冻结的网络模型(pb格式的文件)节点时可用以下的代码实现;
拿ssd-mobilenet v1模型进行试验;
代码部分:
import tensorflow as tf
with tf.Session() as sess:
with open('./ssd_mobilenet_v1_android_export.pb', 'rb') as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
print (graph_def)
前提是你切换到模型所在目录,否则请将上面模型所在位置写成绝对路径。
打印图的部分结果如下:
node {
name: "Postprocessor/BatchMultiClassNonMaxSuppression/map/while/PadOrClipBoxList/cond_1/strided_slice/stack"
op: "Const"
input: "^Postprocessor/BatchMultiClassNonMaxSuppression/map/while/PadOrClipBoxList/cond_1/switch_f"