tensorflow模型打包成pb

多個pb模型合并成一個

1. 查看節點名字

對 tf 不是很熟悉，所以有時間節點名字不清楚，用此辦法找節點名字

# 處在調試狀態，圖已經建立起來后

2. 生成 pb 文件

# output_node_names 為輸出節點的名字

3. inference

主要是利用 get_tensor_by_name 來實現(注：此處節點名字要加上 :0）

with

4. 合并多個圖

用于將多個圖合并到一起，這樣只用一個 sess 就可以得到多個 model 輸出

import

總結

以上是生活随笔為你收集整理的tensorboard ckpt pb 模型的输出节点_tensorflow, ckpt 转 pb 并进行预测的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。