MATLAB构建神经网络识别Mnist手写数字数据集（附代码）

目录

Mnist数据集介绍

构建多层感知机模型

代码：

模型结构：

运行结果：

构建卷积神经网络：

代码：

模型结构：

运行结果：

希望能够对你有所帮助！如有错误之处，敬请指正！

---

Mnist数据集介绍

Mnist数据集中包含60000张共10类手写数字的图片，每张图片的大小为28*28。部分手写数字图片如下所示：

选用其中50000张图片作为训练集，10000张图片作为测试集。

数据集链接：https://pan.quark.cn/s/03a502977f21

构建多层感知机模型

代码：

clearclc%导入数据集matlabroot = './mnist_train_jpg_60000';digitDatasetPath = fullfile(matlabroot);imds = imageDatastore(digitDatasetPath,'IncludeSubfolders',true,'LabelSource','foldernames');%分割数据集numTrainFiles = 5000;[imdsTrain,imdsValidation] = splitEachLabel(imds,numTrainFiles,'randomize');%创建模型layers = [    imageInputLayer([28 28 1],"Name","imageinput")    fullyConnectedLayer(128,'Name',"fc")    batchNormalizationLayer("Name","batchnorm")    reluLayer("Name","relu")    fullyConnectedLayer(64,'Name',"fc_1")    batchNormalizationLayer("Name","batchnorm_1")    reluLayer("Name","relu_1")    fullyConnectedLayer(32,'Name',"fc_2")    batchNormalizationLayer("Name","batchnorm_2")    reluLayer("Name","relu_2")    fullyConnectedLayer(10,"Name","fc_3")    softmaxLayer("Name","softmax")    classificationLayer("Name","classoutput")];%训练参数  options = trainingOptions('sgdm', ...    'InitialLearnRate',0.01, ...    'MaxEpochs',4, ...    'Shuffle','every-epoch', ...    'ValidationData',imdsValidation, ...    'ValidationFrequency',30, ...    'Verbose',false, ...    'Plots','training-progress'); %进行训练，并输出最终结果   net = trainNetwork(imdsTrain,layers,options);    YPred = classify(net,imdsValidation);   YValidation = imdsValidation.Labels;   accuracy = sum(YPred == YValidation)/numel(YValidation);   disp(accuracy);

模型结构：

生成的模型输入为28*28*1大小的图片，有十个输出（与十种数字类型相对应）。模型包含输入层（imageInputLayer）、全连接层（fullyConnectedLayer）、批量归一化层（batchNormalizationLayer ）、激活函数（reluLayer），softmaxLayer层及分类层（classificationLayer）。各层的详细信息与连接顺序图如下所示：

运行结果：

构建卷积神经网络：

代码：

clearclcmatlabroot = './mnist_train_jpg_60000';digitDatasetPath = fullfile(matlabroot);imds = imageDatastore(digitDatasetPath,'IncludeSubfolders',true,'LabelSource','foldernames');numTrainFiles = 5000;[imdsTrain,imdsValidation] = splitEachLabel(imds,numTrainFiles,'randomize');layers = [    imageInputLayer([28 28 1],"Name","imageinput")    convolution2dLayer([3 3],8,"Name","conv","Padding","same")    batchNormalizationLayer("Name","batchnorm")    reluLayer("Name","relu")    maxPooling2dLayer([5 5],"Name","maxpool","Padding","same")    convolution2dLayer([3 3],16,"Name","conv_1","Padding","same")    batchNormalizationLayer("Name","batchnorm_1")    reluLayer("Name","relu_1")    fullyConnectedLayer(10,"Name","fc_3")    softmaxLayer("Name","softmax")    classificationLayer("Name","classoutput")];  options = trainingOptions('sgdm', ...    'InitialLearnRate',0.01, ...    'MaxEpochs',4, ...    'Shuffle','every-epoch', ...    'ValidationData',imdsValidation, ...    'ValidationFrequency',30, ...    'Verbose',false, ...    'Plots','training-progress');    net = trainNetwork(imdsTrain,layers,options);    YPred = classify(net,imdsValidation);   YValidation = imdsValidation.Labels;   accuracy = sum(YPred == YValidation)/numel(YValidation);   disp(accuracy);

模型结构：

运行结果：

希望能够对你有所帮助！如有错误之处，敬请指正！