《OpenCV系列教程》
项目位置:OpenCV-Sample
代码位置:26-ShowMnist.py
在学习手写图片识别的时候,总想看看那些训练图片什么样子,这个demo就是显示里面的部分图片的内容的。
如果整体下载下面是可以带有MNIST数据包的。
from tensorflow.examples.tutorials.mnist import input_data
import numpy as np
import cv2
mnist = input_data.read_data_sets("MNIST_data", one_hot=True)
images = []
img = []
for i in range(mnist.train.images.shape[0]):
im = mnist.train.images[i]
im = im.reshape(28, 28)
if(img == []):
img = im
else :
#横向组合
img = np.hstack((img, im))
#每行显示60个数字图片
if (img.shape[1] / 28 == 60):
if(images == []):
images = img.copy()
img = []
else:
#纵向组合
images = np.vstack((images, img))
img = []
if(i == 1200):
break;
cv2.imshow('reSize2', images)
cv2.waitKey()
cv2.destroyAllWindows()
最终显示的部分如下: