tensorflow模型打包成pb
多个pb模型合并成一个
1. 查看节点名字
对 tf 不是很熟悉,所以有时间节点名字不清楚,用此办法找节点名字
# 处在调试状态,图已经建立起来后
![256c7a110df4df68ebbe6e14e84f7eed.png](https://i-blog.csdnimg.cn/blog_migrate/0705083cc1f4557199580424ee44d5db.jpeg)
2. 生成 pb 文件
# output_node_names 为输出节点的名字
3. inference
主要是利用 get_tensor_by_name 来实现(注:此处节点名字要加上 :0)
with
4. 合并多个图
用于将多个图合并到一起,这样只用一个 sess 就可以得到多个 model 输出
import