当前位置:  开发笔记 > 编程语言 > 正文

Keras显示来自数据生成器的图像

如何解决《Keras显示来自数据生成器的图像》经验,为你挑选了1个好方法。

我正在使用像这样的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

我得到正常的颜色,但仍然有一些奇怪的颜色:

在此输入图像描述



1> 小智..:

图像数组的dtype为'float32',只需将其转换为'uint8'即可:

plt.imshow(image.astype('uint8'))

推荐阅读
mylvfamily
这个屌丝很懒,什么也没留下!
DevBox开发工具箱 | 专业的在线开发工具网站    京公网安备 11010802040832号  |  京ICP备19059560号-6
Copyright © 1998 - 2020 DevBox.CN. All Rights Reserved devBox.cn 开发工具箱 版权所有