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 Ansichten (letzte 30 Tage)
Ältere Kommentare anzeigen
nima aalizade
am 16 Dez. 2017
Kommentiert: Mihai Mihaela
am 8 Dez. 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 Kommentare
Akzeptierte Antwort
Weitere Antworten (1)
abdulkader helwan
am 25 Dez. 2017
Hello.. i am having the same problem here. could u please tell me how u solved it if u did so. thanks
4 Kommentare
Siehe auch
Kategorien
Mehr zu Convert Image Type finden Sie in Help Center und File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!