Invalid training data. The output size (5) of the last layer doesn't match the number of classes (5). How to match output size??
3 views (last 30 days)
Show older comments
nima aalizade
on 16 Dec 2017
Commented: Mihai Mihaela
on 8 Dec 2019
net=vgg16();
imds = imageDatastore(fullfile('E:\','data','labels'),...
'IncludeSubfolders',true,'FileExtensions','.dcm','LabelSource','foldernames');
labelCount = countEachLabel(imds);
trainingNumFiles = 105;
rng(1) % For reproducibility
[trainData,testData] = splitEachLabel(imds,...
trainingNumFiles,'randomize');
imageSize = [512 512 1];
numClasses = 5;
encoderDepth = 9;
lgraph = segnetLayers(imageSize,numClasses,encoderDepth);
plot(lgraph)
options = trainingOptions('sgdm','InitialLearnRate',1e-3, ...
'MaxEpochs',50,'VerboseFrequency',10);
seg = trainNetwork(imds,lgraph,options)
0 Comments
Accepted Answer
More Answers (1)
abdulkader helwan
on 25 Dec 2017
Hello.. i am having the same problem here. could u please tell me how u solved it if u did so. thanks
4 Comments
See Also
Categories
Find more on Convert Image Type in Help Center and File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!