GAN Training: Error with augmented image datastore

Question

Nabiilah 2020-8-2

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/574165-gan-training-error-with-augmented-image-datastore

编辑： Walter Roberson 2020-8-3

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

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Walter Roberson 2020-8-2

3
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/574165-gan-training-error-with-augmented-image-datastore#answer_474274

编辑：Walter Roberson 2020-8-3

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

2 个评论
显示无隐藏无

Nabiilah 2020-8-3

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

Nabiilah 2020-8-3

编辑：Nabiilah 2020-8-3

在 MATLAB Online 中打开

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 :)

请先登录，再进行评论。