这里期望把minst的识别结果更加人可识别的显示出来。
- 把minst图片数组保存为原始图片
- 把tensorflow拟合结果打印为数字
然后我们看一看图片和数字是否一致。
保存minst数组为图片
这个需要用到scipy,使用如下命令安装:
pip install scipy
scipy依赖pillow,继续安装:
pip install pillow
然后用下面的代码保存前20张图片:
import input_data
import os
import scipy.misc
minst = input_data.read_data_sets('MINST_data/', one_hot=True)
save_dir = "MINST_data/raw/"
if os.path.exists(save_dir) is False:
os.mkdir(save_dir)
for i in range(20):
image_array = minst.test.images[i, :].reshape(28,28)
file_name = save_dir + 'minst_test_%d.jpg' % i
scipy.misc.toimage(image_array, cmin=0, cmax=1).save(file_name)
打印tf识别结果
和上一篇文章minst识别结果入门(tensorFlow)类似,这里打印出前20张测试图片的识别数字:
conv_eval = y_conv.eval(feed_dict={x: minst.test.images, y_: minst.test.labels, keep_prob: 1}, session=sess)
for i in range(20):
z = 0
for j in conv_eval[i]:
if j > 0.5:
print(i, z)
else:
z = z + 1
现在可以看一下打印出来的数字和图片是不是一样。
别说,有些图片中的数字真难认,人都不一定认识!