我目前正在使用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格式,它们在保存时只是被转换。
通道中的不匹配可能是由于您用于加载和存储图像的库所致。选中它可以帮助您解决此问题。