Skip to content
Advertisement

how to use more three channels input in train_datagen

I am trying to apply Keras for images with more than three spectral channels. I noticed that train_datagen handles images with three channels based on color_mode=’rgb’. Is there any way to increase the number input channels or are there any alternative methods?

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)

Advertisement

Answer

You can have 1, 3 or 4 channels. See the docs.

color_mode One of “grayscale”, “rgb”, “rgba”. Default: “rgb”. Whether the images will be converted to have 1, 3, or 4 channels.

An alternative is to use tf.data.Dataset for more control. There are examples in the official documentation.

Advertisement