Matlab卷积神经网络(CNN)手写数字识别(一)

今天买的书到了,开始接触卷积神经网络,展示书中内容~

Matlab卷积神经网络手写数字识别(一)

机器学习的基本流程

Alt在机器学习中,一般将数据集划分为两大部分:一部分用于模型训练,称作训练集(Train Set);另一部分用于模型泛化能力评估,称作测试集(Test Set)。在模型训练阶段会将训练集再次划分为两部分,一部分用于模型的训练,而另外一部分用于交叉验证,称作验证集(Validation Set)。如图所示。
请添加图片描述

加载Matlab自带数据集

Matlab 2022b自带手写数据集,其中包含0~9十种手写数字,分别以数字0 ~ 9命名的文件夹中,每个文件夹中有1000幅图像。保存路径如图所示。(版本不同,路径可能有所不同,也可在网上下载数据集。)
在这里插入图片描述

  1. 使用imageDatastore()函数加载数据
    语法1
    imds = imageDatastore(location)
    输入:location表示图像样本存储为可供训练和验证的数据。
    输出:imds表示可供训练和验证的数据
    语法2
    imds = imageDatastore(location,Name,Value)
    可以通过指定“名称-取值”对(Name和Value)来配置特定属性(将每种属性名称括在单引号中)具体含义见表。
名称含义
IncludeSubfolders子文件夹包含标志位。指定true表示可包含每个文件夹中的所有文件和子文件夹,指定 false则表示仅包含每个文件夹中的文件
LabelSource提供标签数据的源。如果指定为’none’,则Labels属性为空;如果指定了’foldernames’, 将根据文件夹名称分配标签并存储在Labels属性中
  • 创建路径
filename='D:\Program Files\MATLAB\R2022b\toolbox\nnet\nndemos\nndatasets\DigitDataset';
  • 将该路径下的图像转化为可用的训练及验证数据集
imds=imageDatastore(filename,...
    "IncludeSubfolders",true,...%包括路径下的所有文件和子文件
    "LabelSource","foldernames");%将文件夹的名称作为标签
  • 随机显示数据集图像
perm=randperm(10000,16);
for i=1:16
    subplot(4,4,i);
    imshow(imds.Files{perm(i)});
    title(imds.Labels(perm(i)));
end
  • 结果
    在这里插入图片描述
    加载成功~
  • 4
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
数字识别是计算机视觉领域中的一个重要应用,它可以广泛应用于自动化识别、物联网智能监控、人机交互等方面。本文介绍基于卷神经网络(Convolutional Neural Network, CNN)的数字识别MATLAB实现。 1. 数据集准备 我们使用的是MNIST数字数据集,它包含了60000张训练图像和10000张测试图像。我们需要将这些图像转换为MATLAB可读取的形式。可以通过以下代码实现: ```matlab % 加载MNIST数据集 train_images = loadMNISTImages('train-images-idx3-ubyte'); % 60000x784 double train_labels = loadMNISTLabels('train-labels-idx1-ubyte'); % 60000x1 double test_images = loadMNISTImages('t10k-images-idx3-ubyte'); % 10000x784 double test_labels = loadMNISTLabels('t10k-labels-idx1-ubyte'); % 10000x1 double % 将图像数据reshape为28x28的矩阵 train_images = reshape(train_images, [28, 28, 1, 60000]); test_images = reshape(test_images, [28, 28, 1, 10000]); ``` 2. CNN模型构建 我们使用MATLAB自带的Deep Learning Toolbox构建CNN模型。以下是一个简单的CNN模型示例: ```matlab layers = [ imageInputLayer([28 28 1]) % 输入层,28x28x1图像 convolution2dLayer(5, 20) % 卷层,5x5核,20个输出通道 reluLayer % 非线性激活函数 maxPooling2dLayer(2, 'Stride', 2) % 池化层,2x2最大池化,步长为2 fullyConnectedLayer(10) % 全连接层,10个神经元输出 softmaxLayer % Softmax激活函数 classificationLayer % 分类层 ]; ``` 3. 模型训练与测试 我们使用MATLAB自带的trainNetwork函数进行模型训练,并使用测试集对模型进行评估。以下是训练和测试代码示例: ```matlab % 设置训练选项 options = trainingOptions('sgdm', ... 'MaxEpochs', 10, ... 'InitialLearnRate', 0.001, ... 'Verbose', false, ... 'Plots', 'training-progress'); % 训练模型 net = trainNetwork(train_images, categorical(train_labels), layers, options); % 使用测试集评估模型 pred_labels = classify(net, test_images); accuracy = sum(pred_labels == categorical(test_labels)) / numel(test_labels); fprintf('Test accuracy: %.2f%%\n', accuracy * 100); ``` 经过10个epoch的训练,模型在测试集上取得了大约98%的准确率。 4. 结论 本文介绍了基于卷神经网络数字识别MATLAB实现,包括数据集准备、模型构建、训练和测试。读者可以基于这个示例代码深入学习CNN模型及其应用。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值