pb文件生成event... 用tensorboard查看

最新推荐文章于 2024-07-05 14:49:33 发布

原创最新推荐文章于 2024-07-05 14:49:33 发布 · 657 阅读

0 ·

CC 4.0 BY-SA版权

总结专栏收录该内容

45 篇文章

订阅专栏

博客提及将PB文件进行转储的事件，涉及信息技术中文件处理相关内容。

将PB文件转储我的.pb文件事件

import tensorflow as tf
from tensorflow.python.platform import gfile

INCEPTION_LOG_DIR = '/tmp/inception_v3_log'

if not os.path.exists(INCEPTION_LOG_DIR):
    os.makedirs(INCEPTION_LOG_DIR)
with tf.Session() as sess:
    model_filename = './model/tensorflow_inception_v3_stripped_optimized_quantized.pb'
    with gfile.FastGFile(model_filename, 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        _ = tf.import_graph_def(graph_def, name='')
    #writer = tf.train.SummaryWriter(INCEPTION_LOG_DIR, graph_def)
    writer=tf.summary.FileWriter(INCEPTION_LOG_DIR, graph_def)                                                
    writer.close()