使用OpenCV或Matplotlib/Pyplot可视化MNIST数据集

我有MNIST数据集，并尝试使用pyplot进行可视化。数据集采用cvs格式，每行是一个由784个像素组成的图像。我希望在pyplot或opencv中以28*28的图像格式进行可视化。我直接尝试使用以下代码：

plt.imshow(X[2:],cmap =plt.cm.gray_r, interpolation = "nearest")

但它不起作用？你有什么建议吗？

回答：

假设你有一个这样的CSV文件，这是MNIST数据集可用的格式

label, pixel_1_1, pixel_1_2, ...

以下是如何在Python中使用Matplotlib和OpenCV进行可视化的方法

Matplotlib / Pyplot

你可以使用上面pixels的numpy数组，它的dtype='uint8'（无符号8位整数）和形状为28 x 28，并使用cv2.imshow()进行绘图

    title = '标签是 {label}'.format(label=label)
    cv2.imshow(title, pixels)
    cv2.waitKey(0)
    cv2.destroyAllWindows()