pb文件tensorboard可視化、查看分析

步驟1:從pb文件中恢復計算圖

import tensorflow as tf

model = 'model.pb'    #pb文件名稱
graph = tf.get_default_graph()
graph_def = graph.as_graph_def()
graph_def.ParseFromString(tf.gfile.FastGFile(model, 'rb').read())
tf.import_graph_def(graph_def, name='graph')
summaryWriter = tf.summary.FileWriter('log/', graph)   #log存放地址

步驟2:tensorboard查看計算圖

命令行中輸入

tensorboard --logdir log/    #log/ 爲log存放地址


 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章