關於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)
結果如下: