我正在使用像这样的keras图像生成器:
val_generator = datagen.flow_from_directory( path+'/valid', target_size=(224, 224), batch_size=batch_size,) x,y = val_generator.next() for i in range(0,1): image = x[i] plt.imshow(image.transpose(2,1,0)) plt.show()
这显示错误的颜色:
我有两个问题.
如何解决问题
如何获取文件的文件名(以便我可以自己从matplotlib这样的东西读取)
编辑:这就是我的datagen的样子
datagen = ImageDataGenerator( rotation_range=3, # featurewise_std_normalization=True, fill_mode='nearest', width_shift_range=0.2, height_shift_range=0.2, horizontal_flip=True )
编辑2:
遵循Marcin的回答后:
image = 255 - image
我得到正常的颜色,但仍然有一些奇怪的颜色:
图像数组的dtype为'float32',只需将其转换为'uint8'即可:
plt.imshow(image.astype('uint8'))