恢复tensorflow模型并将其用于单个输入

时间:2017-03-19 07:06:09

标签: machine-learning tensorflow neural-network

在尝试从此repo im2latex-tensorflow获取训练模型时,我从repo中提供的ckpt files下载了检查点文件

对机器学习不太了解。只想看看输出。经过一段谷歌搜索后,我尝试恢复模型和图形。

from PIL import Image
import tensorflow as tf
import numpy as np
import random
import os

img_input='60ee89c612.png'

new_saver = tf.train.import_meta_graph('weights_best.ckpt.meta')

sess = tf.Session(config=tf.ConfigProto(intra_op_parallelism_threads=8))

new_saver.restore(sess, "./weights_best.ckpt")

sess.run(tf.global_variables_initializer())

此后不知道该怎么办。我如何给img_input 作为程序的输入并获得相应的乳胶代码。 要使用已恢复的模型我理解需要 输入节点和输出节点以预测输出。 非常感谢帮助

0 个答案:

没有答案