创建Keras输入数组,numpy.append无法正常工作

时间:2018-07-19 13:30:03

标签: python arrays numpy keras append

我正在尝试为功能性Keras模型创建输入数组。我在单个np数组中收集了一组图像,因此该数组的形状为:(nr_images,img_width,img_height,nr_channels)

我使用以下代码:

files = glob.glob ("data/train/part2/*.png")
for myFile in files:
    image = cv2.imread (myFile)
    image=cv2.resize(image,(256,256))
    train.append (image)
train = np.array(train,dtype='float32')
np.save('train',train)

所得数组维数为(426, 256, 256, 3)。这样看来行得通。

但是,如果我通过以下方式查看存储在数组中的图像:

image_train=np.load("train.npy")
image=image_train[0]       #Look at the first image
img = Image.fromarray(image,'RGB')
img.show()

我得到垃圾:Example

我的Keras结果非常糟糕,因此我怀疑这与输入有关。

我做错什么了吗?

0 个答案:

没有答案