Invalid training data. The output size (8) of the last layer does not match the number of classes (6).

Question

AYUSH ANAND SAGAR 2020-9-18

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/596065-invalid-training-data-the-output-size-8-of-the-last-layer-does-not-match-the-number-of-classes-6

回答： Mohammad Sami 2020-9-19

clc;
close all;
clear all;
%% load network
net=alexnet;
ppl=net.Layers;
net=net.Layers(1:19);
% layers(23)=fullyConnectedLayer(7);
% layers(25)=classificationLayer;
layers=[net
    fullyConnectedLayer(8)
    softmaxLayer()
    classificationLayer()];
    
matlabpath='C:\Users\ayush\MATLAB_CAPSTONE\Dataset1';
data=fullfile(matlabpath,'trainingset');
train=imageDatastore(data,'IncludeSubfolders',true,'FileExtensions','.jpg','LabelSource','foldernames');
[imgtrain,imgtest]=splitEachLabel(train,0.8,'randomized');
count=train.countEachLabel;
%% training
opt=trainingOptions('sgdm','MaxEpochs',2,'InitialLearnRate',0.001,'Plots','training-progress','MiniBatchSize',64);
TrainNet=trainNetwork(train,layers,opt);
%% accuracy
pred=classify(TrainNet,imgtest);
accuracy=mean(pred==imgtest.Labels);
    
    

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

AYUSH ANAND SAGAR 2020-9-18

Error using trainNetwork (line 170)

Invalid training data. The output size (8) of the last layer does not match the number of classes (6).

Error in transfer (line 27)

TrainNet=trainNetwork(train,layers,opt);

this is the error thats popping up all the time

请先登录，再进行评论。

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

Answer 1

Mohammad Sami 2020-9-19

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/596065-invalid-training-data-the-output-size-8-of-the-last-layer-does-not-match-the-number-of-classes-6#answer_497086

Your final layer has 8 outputs, however your image data store only has 6 labels / classes. If you expect there to be 8 classes, check you image data store. If there are 6 classes, then change your final layer to have 6 outputs.