我在可视化我的数据集的时候，我发现有些不同
比如我同样的代码可视化CIFAR10数据集是没有问题的
但是我去可视化MINST这种没有RGB数据集的时候，我发现用Pytorch的loader不能直接可视化
我发现，这里面可视化我们需要np.squeeze
squeeze 函数：从数组的形状中删除单维度条目，即把shape中为1的维度去掉
所以如果我们的图片为28x28x1，我们可以用这个函数直接可视化图片
>>> a = e.reshape(1,1,10) >>> a array([[[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]]]) >>> np.squeeze(a) array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
import numpy as np x = np.array([[[0], [1], [2]]]) print(x.shape) d = np.squeeze(x) # 从数组的形状中删除单维条目，即把shape中为1的维度去掉 print(d.shape)
(1,3,1)
(3,)
