码迷,mamicode.com
首页 > 编程语言 > 详细

吴裕雄 python 神经网络——TensorFlow pb文件保存方法

时间:2019-05-16 14:01:24      阅读:23      评论:0      收藏:0      [点我收藏+]

标签:mic   oba   combine   util   ram   tensor   ant   run   lob   

import tensorflow as tf
from tensorflow.python.framework import graph_util

v1 = tf.Variable(tf.constant(1.0, shape=[1]), name = "v1")
v2 = tf.Variable(tf.constant(2.0, shape=[1]), name = "v2")
result = v1 + v2

init_op = tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(init_op)
    graph_def = tf.get_default_graph().as_graph_def()
    output_graph_def = graph_util.convert_variables_to_constants(sess, graph_def, [add])
    with tf.gfile.GFile("E:\\Saved_model\\combined_model.pb", "wb") as f:
           f.write(output_graph_def.SerializeToString())

技术图片

from tensorflow.python.platform import gfile

with tf.Session() as sess:
    model_filename = "E:\\Saved_model\\combined_model.pb"
   
    with gfile.FastGFile(model_filename, rb) as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
    result = tf.import_graph_def(graph_def, return_elements=["add:0"])
    print(sess.run(result))

技术图片

 

吴裕雄 python 神经网络——TensorFlow pb文件保存方法

标签:mic   oba   combine   util   ram   tensor   ant   run   lob   

原文地址:https://www.cnblogs.com/tszr/p/10875157.html

(0)
(0)
   
举报
评论 一句话评论(0
0条  
登录后才能评论!
© 2014 mamicode.com 版权所有 京ICP备13008772号-2
迷上了代码!