tensorflow模型打包成pb

多個pb模型合并成一個

1. 查看節(jié)點(diǎn)名字

對 tf 不是很熟悉，所以有時間節(jié)點(diǎn)名字不清楚，用此辦法找節(jié)點(diǎn)名字

# 處在調(diào)試狀態(tài)，圖已經(jīng)建立起來后

2. 生成 pb 文件

# output_node_names 為輸出節(jié)點(diǎn)的名字

3. inference

主要是利用 get_tensor_by_name 來實(shí)現(xiàn)(注：此處節(jié)點(diǎn)名字要加上 :0）

with

4. 合并多個圖

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

import

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。