How to fix the error using minibatchqueue while using GAN?

Question

santhosh kumar buddepu 2023-8-18

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2009977-how-to-fix-the-error-using-minibatchqueue-while-using-gan

评论： santhosh kumar buddepu 2023-9-13

Hello mathworks team,

I have tried to implement "TrainGenerativeAdversarialNetworkGANExample" using my dataset, which consists of 3 classes (.png images) with total of 1426 images, and each images size is 417*532*3. I am getting an error like as shown below.

I didn't understand how to fix this error. I tried with inbuilt matlab example using "flowers datset", it is working fine. Please help me in this regard.

3 个评论
显示 1更早的评论隐藏 1更早的评论

Rohit 2023-8-28

Hi Santhosh,

Could you share your code and dataset with me, so that I can perform the operations on my end and provide you with further assistance?

santhosh kumar buddepu 2023-8-28

ground_truth.zip

Thank you for reply sir,

I have used the expample given by Matlab

https://in.mathworks.com/help/deeplearning/ug/train-generative-adversarial-network.html?s_eid=PSM_15028#TrainGenerativeAdversarialNetworkGANExample-9

Instead of flower dataset, I have given my dataset named as 'ground_truth'

imds = imageDatastore('ground_truth', ...

'IncludeSubfolders',true, ...

'LabelSource','foldernames');

remaining is same, and Iam getting above error.

Herewith I have attached part of my dataset due to size issues. It contains 3 classes.

请先登录，再进行评论。

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

Answer 1

Milan Bansal 2023-9-5

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/2009977-how-to-fix-the-error-using-minibatchqueue-while-using-gan#answer_1301351

在 MATLAB Online 中打开

Hi,

I understand that you getting an error when running the "minibatchqueue" function of Deep-Learning Toolbox while implemetenting "TrainGenerativeAdversarialNetworkGANExample" with your data set.

In "minibatchqueue" function the number of mini-batch variables must be less than or equal to the number of outputs returned by the "MiniBatchFcn". In this case the "MiniBatchFcn" is the custom preprocessing function defined at the end of the script.

function X = preprocessMiniBatch(data) 
% Concatenate mini-batch 
X = cat(4,data{:}); 
% Rescale the images in the range [-1 1]. 
X = rescale(X,-1,1,InputMin=0,InputMax=255); 
end 

This function returns only 1 output i.e. "X", whereas the number of mini-batch variables in the "minibatchqueue" is 2.

Add the optional argument “numOutputs” in the "minibatchqueue" and set its value as 1 to resolve the issue.

mbq = minibatchqueue(augimds, ... 
    MiniBatchSize=miniBatchSize, ... 
    PartialMiniBatch="discard", ... 
    MiniBatchFcn=@preprocessMiniBatch, ... 
    MiniBatchFormat="SSCB",numOutputs = 1); 