python - Keras ImageDataGenerator 在保存到目录时无意中更改了图像的颜色

Question

我正在尝试使用 ImageDataGenerator 对象的流方法获取输入图像并保存该图像的 10 个增强版本。问题是它无意中改变了图像的颜色，即使我没有向 ImageDataGenerator 类传递任何参数也是如此。

这是我的输入图像和代码，下面是我的输出图像。我正在使用 Tensorflow 2.2.0

input image

from tensorflow.keras.preprocessing import image
from tensorflow.keras.preprocessing.image import ImageDataGenerator

img = image.load_img(r'C:\Users\me\Desktop\test.jpg')
img_array = image.img_to_array(img)
img_array = img_array.reshape((1,) + img_array.shape)

datagen = ImageDataGenerator() #no arguments, no augmentations
save_to_dir = r'C:\Users\me\Desktop'

i = 0
for batch in datagen.flow(img_array, batch_size=1, save_to_dir=save_to_dir, save_format='jpg'):
    if i == 9:
        break
    i += 1

output image

图像的颜色完全不同。任何帮助将不胜感激。

Answer 1

问题是 ImageDataGenerator.flow 方法会自动重新缩放您的图像，而无法在使用 save_to_dir 参数时将其禁用。我能够使用以下代码解决此问题。

from tensorflow.keras.preprocessing import image
from tensorflow.keras.preprocessing.image import ImageDataGenerator

img = image.load_img(r'C:\Users\me\Desktop\test.jpg')
img_array = image.img_to_array(img)
img_array = img_array.reshape((1,) + img_array.shape)

datagen = ImageDataGenerator() #no arguments, no augmentations
save_to_dir = r'C:\Users\me\Desktop'

i = 0
for batch in datagen.flow(img_array, batch_size=1, save_format='jpg'):
    img_save = image.array_to_img(batch[0], scale=False) #scale=False did the trick
    img_save.save(save_to_dir + fr'\augment_{i}.jpg') #save image manually
    if i == 9:
        break
    i += 1