Tensorflow中保存成pb file 需要 使用函數(shù)

graph_util.convert_variables_to_constants(sess, sess.graph_def,

output_node_names=[]) []中需要填寫你需要保存的結(jié)點(diǎn)。如果保存的結(jié)點(diǎn)在神經(jīng)網(wǎng)絡(luò)中沒有被顯示定義該怎么辦？

例如我使用了tf.contrib.slim或者keras，在tf的高層很多情況下都會(huì)這樣。

在寫神經(jīng)網(wǎng)絡(luò)時(shí)，只需要簡單的一層層傳導(dǎo)，一個(gè)slim.conv2d層就包含了kernal，bias，activation function，非常的方便，好處是網(wǎng)絡(luò)結(jié)構(gòu)一目了然，壞處是什么呢？

在嘗試保存pb的 output node names時(shí)，需要將最后的輸出結(jié)點(diǎn)保存下來，與這個(gè)結(jié)點(diǎn)相關(guān)的，從輸入開始，經(jīng)過層層傳遞的嵌套函數(shù)或者操作的相關(guān)結(jié)點(diǎn)，都會(huì)被保存，但無效的例如 計(jì)算準(zhǔn)確率，計(jì)算loss等，就可以省略了，因?yàn)楸４娴膒b主要是用來做預(yù)測的。

在準(zhǔn)備查看所有的結(jié)點(diǎn)名稱并選取保存時(shí)，發(fā)現(xiàn)scope "local3"里面僅有相關(guān)的weights 和biases，這兩個(gè)是單獨(dú)存在的，即保存這兩個(gè)參數(shù)并沒有任何意義。

那么這時(shí)候有兩種解決辦法：

方法一：

graph_util.convert_variables_to_constants(sess, sess.graph_def, output_node_names=[var.name[:-2] for var in tf.global_variables()])

那么這個(gè)的意思是所有的variable的都被保存下來 但函數(shù)中要求的是 node name 我們通過 global_variables獲得的是 變量名 并不是 節(jié)點(diǎn)名

（例如 output:0 就是變量名，又叫tensor name）

output就是 node name了。

在tensorboard中可以一窺究竟

通過這樣 也可以將 所有的變量全部保存下來（但是你并不能使用，是因?yàn)槟愕膐utput并沒有名字，所以你不可以通過常用的sess.graph.get_tensor_by_name來使用）

方法二：

那就是直接改寫神經(jīng)網(wǎng)絡(luò)了....當(dāng)然了還是比較簡單的，只要改寫最后一個(gè)，改寫成output即可，tensorflow中無論是 變量、操作op、函數(shù)、都可以命名，那么這個(gè)地方是一個(gè)簡單的全連接，僅需要將weights*net(上一層的輸出) +bias 即可，我們只要將bias相加的結(jié)果命名為 ouput即可：

with tf.name_scope('local3'):
 
  local3_weights = tf.Variable(tf.truncated_normal([4096, self.output_size], stddev=0.1))
 
  local3_bias = tf.Variable(tf.constant(0.1, shape=[self.output_size]))
 
result = tf.add(tf.matmul(net, local3_weights), local3_bias, name="output")

這樣將上述的convert_variables_to_constants中的output_node_names只需要填寫一個(gè)['output']即可，因?yàn)檫@一個(gè)output結(jié)點(diǎn)，需要從input開始，將所有的神經(jīng)網(wǎng)絡(luò)前向傳播的操作和參數(shù)全部保存下來，因此保存的結(jié)點(diǎn)數(shù)量 和 方法一保存的結(jié)點(diǎn)數(shù)量是一樣的（console顯示都是 convert 24）。

完整的pb保存為:(我是將ckpt讀入進(jìn)來，然后存成pb的)

from tensorflow.python.platform import gfile
 
 
 
load_ckpt():
 
  path = './data/output/loss1.0/'
 
  print("read from ckpt")
 
  ckpt = tf.train.get_checkpoint_state(path)
 
  saver = tf.train.Saver()
 
  saver.restore(sess, ckpt.model_checkpoint_path)
 
 
 
def write2pb_file():
 
  constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def,
 
    output_node_names=["output"])
 
  with tf.gfile.GFile(path+'loss1.0.pb', mode='wb') as f:
 
  f.write(constant_graph.SerializeToString())
 
  print("Model is saved as " + path+'loss1.0.pb')
 
 
 
def main():
 
  load_ckpt()
 
  write2pb_file()
 

如果是簡單的直接保存，那就更簡單了。

pb文件的read，很多人會(huì)將一個(gè)net寫成一個(gè)類，在引入的時(shí)候會(huì)將新建這個(gè)類，然后讀入ckpt文件，這完全沒有問題，但是在讀取pb時(shí)，就會(huì)發(fā)生問題，因?yàn)閜b中已經(jīng)包含了圖與參數(shù)，引入時(shí)會(huì)創(chuàng)建一個(gè)默認(rèn)的圖，但是net類中自己也會(huì)創(chuàng)建一個(gè)圖，那么這時(shí)候你運(yùn)行程序，參數(shù)其實(shí)并沒有使用.pb的文件。

所以我們不能創(chuàng)建net類，然后直接讀入.pb文件，對(duì).pb文件，通過如下代碼，獲取.pb的graph中的輸入和輸出。

self.output = self.sess.graph.get_tensor_by_name("output:0")
 
self.input = self.sess.graph.get_tensor_by_name("images:0")

注意此時(shí)要加：0 因?yàn)槟惬@取的不再是結(jié)點(diǎn)了，而是一個(gè)真實(shí)的變量，我的理解是，結(jié)點(diǎn)相當(dāng)于一個(gè)類，:0是對(duì)象，默認(rèn)初始化值就是對(duì)象的初始化。

然后就可以通過self.sess.run(self.output(feed_dict={self.input: your_input})))運(yùn)行你的網(wǎng)絡(luò)了！

以上這篇tensorflow沒有output結(jié)點(diǎn),存儲(chǔ)成pb文件的例子就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

最新評(píng)論