我目前正在使用 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 格式,它们只是在保存时被转换。