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

【问题标题】：Color mode in Image Data Generator method flow_from_directory图像数据生成器方法 flow_from_directory 中的颜色模式
【发布时间】：2021-04-29 22:25:15
【问题描述】：

我有一个关于 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')

【问题讨论】：

标签： machine-learning deep-learning conv-neural-network resnet transfer-learning

【解决方案1】：

如果你看到文件，它就是：

color_mode：“灰度”、“rgb”、“rgba”之一。默认值：“RGB”。图像是否将转换为 1、3 或 4 通道。

所以，是的，结果是灰度。

【讨论】：

那为什么我们必须在一行中传递 3 个通道：ResNet = InceptionResNetV2(input shape=IMAGE_SIZE + [3] ) 。我们可以在这里传递 [1] ，然后只需要 1 个频道。
ResNet 实现了 3 通道输入。所以你应该把它当作 3 通道图像。你不应该让你的图像灰度。 test_set = test_datagen.flow_from_directory, target_size = (224, 224), batch_size = 32, class_mode = 'categorical',color_mode='rgb')