GAN Training: Error with augmented image datastore

Hi, I had this script running with one dataabse, however the generated images were far too pixelated, so I decided to switch the database to another(and nothing else) and now I'm getting the below error. I'm new to this, could I please have some help debugging it?

Error using augmentedImageDatastore/applyAugmentationPipeline (line 391)
augmentedImageDatastore cannot form MiniBatches of data because input image sizes differ in 3rd
dimension. Consider using 'ColorPreprocessing' option to ensure all augmented images have same
number of channels.

Error in augmentedImageDatastore>@(c)self.applyAugmentationPipeline(c) (line 372)
                Xout = cellfun(@(c) self.applyAugmentationPipeline(c),X,'UniformOutput',false);

Error in augmentedImageDatastore/applyAugmentationPipelineToBatch (line 372)
                Xout = cellfun(@(c) self.applyAugmentationPipeline(c),X,'UniformOutput',false);

Error in augmentedImageDatastore/read (line 308)
            input = self.applyAugmentationPipelineToBatch(input);

Error in GANUTKFace (line 168)
        data = read(augimds);

I've used the script from this example: https://uk.mathworks.com/help/deeplearning/ug/train-generative-adversarial-network.html

Answer 1

[Walter Roberson]

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

You presently have a mix of rgb and grayscale images and need to be consistent; the option can automatically convert

[Nabiilah]

Thank you for clarifying that, where in the script would I specify this option please?

[Nabiilah]

I got it to work by adding the ColorReprocessing option to the below:

%Augment the data to include random horizontal flipping and to resize the image to 64x64
augmenter = imageDataAugmenter('RandXReflection', true);
augimds = augmentedImageDatastore([64 64], imds,'ColorPreprocessing', 'gray2rgb', 'DataAugmentation', augmenter);

Thank you :)