python - 从 cifar-10 数据集加载图像

Question

我正在使用 cifar-10 数据集来训练我的分类器。我已经下载了数据集并尝试显示数据集中的图像。我使用了以下代码:

from six.moves import cPickle as pickle
from  PIL import Image
import numpy as np

f = open('/home/jayanth/udacity/cifar-10-batches-py/data_batch_1', 'rb')

tupled_data= pickle.load(f, encoding='bytes')

f.close()

img = tupled_data[b'data']

single_img = np.array(img[5])

single_img_reshaped = single_img.reshape(32,32,3)

plt.imshow(single_img_reshaped)

数据说明如下:每个数组存储一个 32x32 彩色图像。前 1024 个条目包含红色 channel 值，接下来的 1024 个条目包含绿色 channel 值，最后 1024 个条目包含蓝色 channel 值。图像以行优先顺序存储，因此数组的前 32 个条目是图像第一行的红色 channel 值。

我的实现是否正确？

上面的代码给了我下面的图像:

Answer 1

我用过

single_img_reshaped = np.transpose(np.reshape(single_img,(3, 32,32)), (1,2,0))

在我的程序中获得正确的格式。