文件目录如下:
恢复模型
import tensorflow as tf
import numpy as np
#with tf.Session() as sess:
sess = tf.Session()
saver = tf.train.import_meta_graph('model/gesture.meta')
saver.restore(sess, tf.train.latest_checkpoint('./model'))
#获取输入和输出tensor
x = sess.graph.get_tensor_by_name('inputs:0')
p = sess.graph.get_tensor_by_name('predict:0')
print(sess.run(p, feed_dict={x: np.random.randint(0, 2, [10, 128, 64, 3], dtype=np.uint8)}))