matlab - 无效的训练数据神经网络 Matlab

Question

我的代码不断收到此错误，之前它运行良好，但我不确定我做了什么，我设法得到了这个错误：

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

Error in Main_Demo (line 64) CNN_net = trainNetwork(im_datastoreTrain,CNNl

这是我的代码：

input_layerSize = [62 50 1];
Total_Classes = 2;

CNNlayers = [
    imageInputLayer(input_layerSize)
    convolution2dLayer(5,20)
    batchNormalizationLayer
    reluLayer
    fullyConnectedLayer(Total_Classes)
    softmaxLayer
    classificationLayer];

%% 5- Network parameters definition
CNN_parameters = trainingOptions('sgdm', ...
    'MaxEpochs',4, ...
    'ValidationData',im_datastoreTest, ...
    'ValidationFrequency',45, ...
    'Verbose',false, ...
    'Plots','training-progress');
%% 6 - Training of the Network 
CNN_net =  trainNetwork(im_datastoreTrain,CNNlayers,CNN_parameters);

如果有人知道如何解决此错误，请告诉我，谢谢。

标签： matlab