建立与读取.pb文件

最新推荐文章于 2024-07-21 07:15:00 发布

原创

最新推荐文章于 2024-07-21 07:15:00 发布 · 6.1k 阅读

收录于

本文详细介绍了如何创建和读取protobuf序列化后的.pb文件，包括protobuf的基本概念、序列化与反序列化过程，以及在实际项目中使用.pb文件进行数据交换的方法。

#coding=utf-8
import tensorflow as tf
from tensorflow.python.framework import graph_util

x = tf.placeholder(shape=[1], dtype=tf.float32, name='x')

varibale_1 = tf.get_variable('v1', [1], tf.float32, initializer=tf.random_normal_initializer(mean=1))

output = tf.multiply(x, varibale_1, name='mul')

initial_op = tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(initial_op)
    graph_def = tf.get_default_graph().as_graph_def()#将图定义取出
#     print(graph_def)
    out_graph = graph_util.convert_variables_to_constants(sess, graph_def, ['mul'])#将图中的变量转化为constant
    print(sess.run(output,{x:[5]}))
    print(sess.run(varibale_1))
    with tf.gfile.GFile('./model.pb','wb') as f:
        f.write(out_graph.SerializeToString())#将图定义转化为字符串形式并且写入.pb文件中