How to fix the error using minibatchqueue while using GAN?
显示 更早的评论
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 个评论
santhosh kumar buddepu
2023-8-26
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
采纳的回答
Milan Bansal
2023-9-5
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);
Refer to documentation link to know more about "minibatchqueue" function.
更多回答（0 个）
类别
在 帮助中心 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!
Translated by 
