첨부 실행 코드는 나눔고딕코딩 폰트를 사용합니다.
유용한 소스 코드가 있으면 icodebroker@naver.com으로 보내주시면 감사합니다.
블로그 자료는 자유롭게 사용하세요.

728x90
반응형
import random
import tensorflow as tf
import tensorflow.examples.tutorials.mnist as mnist

mnistDatasets = mnist.input_data.read_data_sets("data", one_hot = True)

session = tf.InteractiveSession()

saver = tf.train.import_meta_graph("data/mnist_1_layer_softmax.ckpt.meta")

saver.restore(session, "data/mnist_1_layer_softmax.ckpt")

tf.get_default_graph().as_graph_def()

inputTensor  = session.graph.get_tensor_by_name("input:0" )
outputTensor = session.graph.get_tensor_by_name("output:0")

testImageIndex   = random.randint(0, mnistDatasets.test.images.shape[0])
testImageNDArray = mnistDatasets.test.images[testImageIndex]

result = session.run(["input:0", outputTensor], feed_dict = {inputTensor : [testImageNDArray]})

print("정답 : ", session.run(tf.argmax(mnistDatasets.test.labels[testImageIndex])))
print("판단 : ", session.run(tf.argmax(result[1], 1)[0]))
728x90
반응형
Posted by 사용자 icodebroker

댓글을 달아 주세요