TensorFlowで保存済モデルからのVariable抽出
ニューロンとか可視化したいときとかに
import tensorflow as tf # グラフ構造import saver = tf.train.import_meta_graph('models/my_model_final.ckpt.meta') # sessionにグラフ構造をrestore sess = tf.Session() saver.restore(sess, 'models/my_model_final.ckpt') # variablesのリストを取得 tvars = tf.trainable_variables() # 値の取得 vars = sess.run(tvars)