吴裕雄 python 神经网络——TensorFlow pb文件保存方法
2024-09-06 19:18:15
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))
最新文章
- map 容器的使用
- Extjs 百度地图扩展
- Ubuntu 12.04 root账户开启及密码重设
- 樱花漫地集于我心,蝶舞纷飞祈愿相随---总结 适者:survival of the fittest 适者:survival of the fittest
- WPF 之 左键弹出操作菜单,并禁用右键菜单
- WIFI(1)WIFI直连 + socket 可以用来实现类似蓝牙传输数据的功能
- javascript进击(五)JS对象
- VS2015开发的Office Addin部署,安装时报错:无法解析属性“type”的值。
- 360回归A股,周鸿祎来给BAT和小米添堵了
- NodeJS中的http模块
- LeetCode 624. Maximum Distance in Arrays (在数组中的最大距离)$
- [爬虫]Scrapy 错误ordinal not in range(128)
- Django 基础二(View和urls)
- 24访问者模式Visitor
- WPF设置对象隐藏、不可用
- [转] 带你彻底理解RSA算法原理
- 【原】python3.7 无法pip安装提示ssl错误解决方案
- Druid连接池简介和配置
- 《转》python学习(7) -列表
- SqueezeNet:AlexNet-level Accuracy with 50x fewer parameters and less than 0.5Mb model size