mnist识别自己用画板手写的数字

验证训练好的mnist模型

• 用window画板随意写个数字并保存图片

from PIL import Image
#降为单通道
image_data = Image.open('test.png').convert('L')
#resize
if image_data.size[0] != 28 or image_data.size[1] != 28:
    image_data = image_data.resize((28, 28))

arr = np.array(image_data, dtype=np.float32)
arr = 1.0 - arr/255.

plt.imshow(np.array(arr).reshape(28,28), cmap = 'binary')
plt.show()
#拿训练好的模型运行operation 'output'
result = sess.run(output, feed_dict={x: np.array(arr).reshape(1, 784)})

print (result)
print (np.argmax(result))