如何在train_datagen中使用更多的三个通道输入

Question

我试图将Keras应用于具有三个以上光谱通道的图像。我注意到train_datagen处理基于color_mode='rgb'的三个通道的图像。有没有办法增加输入通道的数量或者有其他的方法?

img_train_generator = train_datagen.flow_from_directory(
"train/image/",
target_size=(64, 64),
batch_size=batch_size,
class_mode= None,
color_mode='rgb',
shuffle=False)

Answer 1

可以设置1、3或4个通道。参考文档

color_mode"灰度"、"rgb"、"rgb"中的一种。默认值:"rgb"图像是否转换为1、3或4通道。

另一种选择是使用tf.data.Dataset来进行更多的控制。官方文档中有示例。