Keras图像预处理流程将RGB图像转换为BGR

时间:2019-01-17 12:03:19

标签: python keras rgb bgr

我目前正在使用Keras的图像预处理功能来增强一些训练图像数据。作为此过程的一部分,我试图可视化增强,可以通过使用ImageDataGenerator类中的flow方法将图像保存到目录中来完成:

https://keras.io/preprocessing/image/#flow

datagenerator.flow(image, batch_size=1, save_to_dir=args["imgdir"], save_prefix='aug',
                   save_format='png')

问题是我传递的图像是RGB,保存在目录中的图像是BGR。我正在做的唯一转换是旋转,为什么将它们转换为BGR?通过将图像转换为BGR,然后再将其传递给生成器流方法,可以解决这种情况。

生成器本身不生成BGR图像-那些仍保持RGB格式,它们只是在保存时进行转换。

1 个答案:

答案 0 :(得分:0)

通道不匹配可能是由于您用于加载和存储图像的库所致。选中它可以帮助您解决此问题。