MATLAB Answers

How can i fix this error?

조회 수: 49(최근 30일)
Hello, i have a problem with the size of the training data. i've done the augmentedImageDatastore function to fix it but when i do the trainNetwork function they say that the size of the training images is not the same as the size that the input layer expects. How can i fix this error? Here is my code
Flowers ='C:\Users\hind';'deeplearning_course_files';'Flowers';
flowerds = imageDatastore(Flowers,"IncludeSubfolders",true,"LabelSource","foldernames");
[trainImgs,testImgs] = splitEachLabel(flowerds,0.6);
numClasses = numel(categories(flowerds.Labels));
net = googlenet;
lgraph = layerGraph(net);
newFc = fullyConnectedLayer(12,"Name","new_fc");
lgraph = replaceLayer(lgraph,"loss3-classifier",newFc);
newOut = classificationLayer("Name","new_out");
lgraph = replaceLayer(lgraph,"output",newOut);
options = trainingOptions("sgdm","InitialLearnRate", 0.001);
auds = augmentedImageDatastore([224 224],flowerds,"ColorPreprocessing","gray2rgb");
[flowernet,info] = trainNetwork(trainImgs, lgraph, options);
  댓글 수: 4
Hind Haboubi
Hind Haboubi 19 Mar 2021
thank you, I'll try it

댓글을 달려면 로그인하십시오.

채택된 답변

Srivardhan Gadila
Srivardhan Gadila 22 Mar 2021
Even though you are creating a "auds" augmentedImageDatastore object, you are not using it to train your network. Change the line
[flowernet,info] = trainNetwork(trainImgs, lgraph, options);
to
[flowernet,info] = trainNetwork(auds, lgraph, options);

추가 답변(0개)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by