Image Classification Training Issue

The following error comes while executing the image classification task.

augementedImageDatastore cannot form minibatches of data because the input image sizes differ in the 3rd dimension.Consider using color preprocessing to ensure all augmented images have same no of channels.

Answer 1

[Divit]

Hi Divya,

The error you're encountering indicates that the images in your dataset have different numbers of channels. For example, some images might be grayscale (single channel) while others are RGB (three channels). 'augmentedImageDatastore' expects all images to have the same number of channels.

To resolve this issue, you can preprocess your images using a custom function to ensure they all have the same number of channels.

Alternatively, you can use the 'ColorPreprocessing' property of 'augmentedImageDatastore'. You can learn more about the 'ColorPreprocessing' property by going through the following documentation link:

https://www.mathworks.com/help/deeplearning/ref/augmentedimagedatastore.html#mw_d4cb9c52-c701-4e94-95d8-96ae545d295f

[Divya]

I had applied this command to create the augmented data store but not sure how to save the images in a new folder

[Divit]

Hi,

You can use the 'imwrite' function to save the images after you have completed the preprocessing.

To learn more about 'imwrite', you can refer to the following documentation link: https://www.mathworks.com/help/matlab/ref/imwrite.html