我正在尝试为功能性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()
我得到垃圾:
我的Keras结果非常糟糕,因此我怀疑这与输入有关。
我做错什么了吗?