从cifat10-dataset

时间:2015-11-13 08:38:14

标签: python image dataset

我正在尝试从cifar10-dataset获取图像。当我从数组重建图像时, 我在一张图片中看到了9张相同的图片,我不知道这是什么问题。

当我从数据加载图像时,single_img形状(3072,)。在那之后,我重塑了 我的single_img变量(32,32,3)。我不知道问题出在哪里。 这是我的代码;

import cPickle
from  PIL import Image
import numpy as np

f = open("/home/leo/Downloads/cifar-10-batches-py/data_batch_1", "rb")

tupled_data= cPickle.load(f)

f.close()

img = tupled_data['data']

single_img = np.array(img[0])

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

j = Image.fromarray(single_img_reshaped)

j.save("/home/leo/Desktop/blabla.bmp")

示例图像; enter image description here

1 个答案:

答案 0 :(得分:0)

请务必注意图像像素阵列的格式..

[R....G....B]

所以你只需将其格式更改为

[[[R,G,B],....,[R,G,B]]
[[R,G,B],....,[R,G,B]]
[[R,G,B],....,[R,G,B]]]

但是

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

不要像以前那样去做。