CIFAR-10数据集输出图片

关于CIFAR-10数据集向量到三维图片中的转置操作

在CIFAR-10数据集中，图片的储存方式为一维向量，如下图所示

错误方法：

a = x_data[3,:] # np.ndarray shape:(3072,)
a = a.reshape(32,32,3)
plt.imshow(a)

正确方法：

a = x_data[3,:] # np.ndarray shape:(3072,)
a = a.reshape(3,32,32).transpose(1,2,0)
plt.imshow(a)

首先使用reshape将numpy变为3x32x32的矩阵，

之后将三个维度进行转置（重新排列）为(dim1,dim2,dim0)

结果如下：

标签：10,输出,转置,reshape,CIFAR,shape,32
来源： https://www.cnblogs.com/aminor/p/14333376.html