app.py

#coding:utf-8

import tensorflow as tf
import backward
import forward
import PreProcess as PP


def restore_model(testArr):
	with tf.Graph().as_default() as tg:
		x = tf.placeholder(tf.float32, [None, forward.INPUT_NODE])
		y = forward.forward(x, None)
		preValue = tf.argmax(y, 1)

		variable_averages = tf.train.ExponentialMovingAverage(backward.MOVING_AVERAGE_DECAY)
		variables_to_restore = variable_averages.variables_to_restore()
		saver = tf.train.Saver(variables_to_restore)

		with tf.Session() as sess:
			ckpt = tf.train.get_checkpoint_state(backward.MODEL_SAVE_PATH)
			if ckpt and ckpt.model_checkpoint_path:
				saver.restore(sess, ckpt.model_checkpoint_path)
				preValue = sess.run(preValue, feed_dict={x:testArr})
				return preValue
			else:
				print("No checkpoint file found")
				return -1


def application(file_path):
	data = PP.image_process(file_path)
	lable = ''
	if(len(data)==0):
		print("识别失败，请传入更清晰的图片")
	else:
		print("正在识别......")
		for i in range(len(data)):
			preValue = restore_model(data[i:i + 1])[0]
			lable += str(preValue)
		print("识别结果："+lable)