Tensorflow Android端开发之——模型节点信息查看

最新推荐文章于 2022-02-12 06:16:30 发布

ciky奇

最新推荐文章于 2022-02-12 06:16:30 发布

阅读量1.3k

点赞数

分类专栏： Android TensorFlow 深度学习/机器学习移动端（边缘设备）深度学习文章标签： ssd-mobilenet v1

本文链接：https://blog.csdn.net/c20081052/article/details/84557370

版权

移动端（边缘设备）深度学习同时被 3 个专栏收录

16 篇文章 97 订阅 ¥9.90 ¥99.00

订阅专栏

深度学习/机器学习

81 篇文章 12 订阅

订阅专栏

TensorFlow

56 篇文章 15 订阅

订阅专栏

本文介绍了如何查看TensorFlow冻结的pb模型（如ssd-mobilenet v1）的节点信息，包括使用代码打印节点详细信息和通过TensorBoard进行可视化。通过TensorBoard生成的图结构，可以清晰地找到输出节点名，如：num_detections, detection_scores, detection_boxes，这对于理解模型结构和移动端的推理工作至关重要。" 119492670,11272735,Linux中的基本储存服务详解,"['Linux', '磁盘管理', '文件系统', '分区', '挂载']

摘要由CSDN通过智能技术生成

查看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"

了解本专栏