图像数据生成器方法flow_from_directory中的颜色模式

Question

我有一个关于ImageDataGenerator的flow_from_directory方法的问题。我的图像是RGB形式，我正在使用迁移学习，我想在灰度图像上训练模型，但预训练模型期望3个通道。在flow_from_directory颜色模式中有一个方法，它采用灰度或RBG。我的问题是，如果我在这里写灰度，Keras会自动将RGB图像转换为灰度还是其他东西?

IMAGE_SIZE=[224,224]
resnet = InceptionResNetV2(input_shape=IMAGE_SIZE + [3] , weights='imagenet', include_top=False)

Flow_from_dir方法:

test_set = test_datagen.flow_from_directory(valid_path,
target_size = (224, 224),
batch_size = 32,
class_mode = 'categorical',color_mode='grayscale')

Answer 1

如果你看到文档，它显示:

color_mode: "grayscale" rgb" rgb"默认值:"rgb"图像是否转换为1、3或4通道。

所以，是的，它原来是一个灰度。