当前位置：首页 > news >正文

基于MATLAB的CNN图像分类算法实现

news 2026/5/12 5:46:39

基于MATLAB深度学习工具箱（Deep Learning Toolbox）的卷积神经网络（CNN）图像分类完整实现流程

一、数据准备与预处理

1.数据集加载

使用MATLAB内置的imageDatastore函数加载图像数据集，支持自定义路径或直接调用公开数据集（如CIFAR-10、MNIST）：

% 加载CIFAR-10数据集（示例）[XTrain,YTrain]=digitTrain4DArrayData;% 内置手写数字数据集[XTest,YTest]=digitTest4DArrayData;% 或自定义数据集（需按文件夹分类）dataFolder='path/to/custom_dataset';imds=imageDatastore(dataFolder,...'IncludeSubfolders',true,...'LabelSource','foldernames');

关键点：

图像需按类别存放在子文件夹中，文件夹名即为标签。
对于灰度图像，需转换为RGB格式（如cat(3, X, X, X)）。

2.数据预处理

归一化：将像素值缩放到[0,1]或[-1,1]范围。
数据增强：通过旋转、平移等操作增加数据多样性，防止过拟合：

augmenter=imageDataAugmenter(...'RandRotation',[-20,20],...% 随机旋转±20度'RandXReflection',true,...% 随机水平翻转'RandYReflection',true);% 随机垂直翻转augimdsTrain=augmentedImageDatastore([32,32],imdsTrain,'DataAugmentation',augmenter);

关键点：输入尺寸需与网络输入层匹配（如AlexNet需227×227）。

二、CNN模型构建

1.自定义网络架构

通过layerGraph定义网络结构，典型CNN包含卷积层、池化层、全连接层：

layers=[imageInputLayer([32323])% 输入层（高度×宽度×通道）% 卷积块1convolution2dLayer(3,16,'Padding','same')% 3×3卷积核，输出16通道batchNormalizationLayer% 批归一化reluLayer% ReLU激活maxPooling2dLayer(2,'Stride',2)% 2×2最大池化% 卷积块2convolution2dLayer(3,32,'Padding','same')batchNormalizationLayer reluLayermaxPooling2dLayer(2,'Stride',2)% 全连接层fullyConnectedLayer(10)% 输出层（类别数）softmaxLayer% Softmax概率转换classificationLayer];% 分类输出

优化建议：

添加Dropout层（如dropoutLayer(0.5)）进一步防止过拟合。
使用残差连接（ResNet结构）提升深层网络性能。

2.迁移学习（推荐）

对于小数据集，直接使用预训练模型（如ResNet-50）进行微调：

net=resnet50;% 加载预训练模型lgraph=layerGraph(net);% 替换最后的全连接层和输出层newFCLayer=fullyConnectedLayer(numClasses,'WeightLearnRateFactor',10);newOutputLayer=classificationLayer('Name','new_classoutput');lgraph=replaceLayer(lgraph,'fc1000',newFCLayer);lgraph=replaceLayer(lgraph,'ClassificationLayer_fc1000',newOutputLayer);

优势：迁移学习可显著减少训练时间并提升准确率。

三、模型训练与优化

1.训练选项配置

通过trainingOptions设置优化器、学习率、早停等参数：

options=trainingOptions('adam',...% 优化器（sgdm, rmsprop）'MaxEpochs',20,...'MiniBatchSize',64,...'InitialLearnRate',0.001,...'Shuffle','every-epoch',...'ValidationData',imdsValidation,...% 验证集'ValidationFrequency',30,...'Verbose',false,...'Plots','training-progress',...% 实时显示训练曲线'ExecutionEnvironment','auto');% 自动选择CPU/GPU加速

关键参数：

LearnRateSchedule: 设置学习率衰减策略（如'piecewise'）。
L2Regularization: 添加L2正则化（默认0.0005）。

2.模型训练

调用trainNetwork启动训练：

net=trainNetwork(augimdsTrain,lgraph,options);

训练监控：

实时显示损失（Loss）和准确率（Accuracy）曲线。
若启用验证集，可观察验证损失防止过拟合。

四、模型评估与测试

1.测试集预测

YPred=classify(net,imdsTest);YTest=imdsTest.Labels;accuracy=sum(YPred==YTest)/numel(YTest);fprintf('测试集准确率:%.2f%%',accuracy*100);

2.混淆矩阵分析

figure;cm=confusionchart(YTest,YPred);cm.Title='混淆矩阵';cm.ColumnSummary='column-normalized';% 列归一化cm.RowSummary='row-normalized';% 行归一化

解读：对角线元素为正确分类比例，非对角线反映误分类情况。

3.ROC曲线与AUC

[X,Y,T,AUC]=perfcurve(YTest,scores(:,2),'cat');% 假设二分类figure;plot(X,Y);title(sprintf('ROC曲线 (AUC=%.2f)',AUC));

五、模型部署与应用

1.新图像分类

newImg=imread('test_image.jpg');newImgResized=imresize(newImg,[32,32]);% 调整尺寸label=classify(net,newImgResized);imshow(newImg);title(sprintf('预测类别: %s',char(label)));

2.导出为ONNX格式

net=exportNetworkToONNX(net,'cnn_model.onnx');

应用场景：部署到移动端或嵌入式设备（需转换框架如TensorRT）。

参考代码利用MATLAB实现一个基于CNN的图像分类算法www.youwenfan.com/contentcsq/50981.html

六、完整代码示例

%% 数据准备dataFolder='path/to/dataset';imds=imageDatastore(dataFolder,'IncludeSubfolders',true,'LabelSource','foldernames');[imdsTrain,imdsTest]=splitEachLabel(imds,0.8,'randomized');% 数据增强augmenter=imageDataAugmenter('RandRotation',[-15,15],'RandXReflection',true);augimdsTrain=augmentedImageDatastore([224,224],imdsTrain,'DataAugmentation',augmenter);%% 模型构建（迁移学习）net=alexnet;lgraph=layerGraph(net);newFCLayer=fullyConnectedLayer(10,'WeightLearnRateFactor',10);newOutputLayer=classificationLayer('Name','output');lgraph=replaceLayer(lgraph,'fc7',newFCLayer);lgraph=replaceLayer(lgraph,'ClassificationLayer_fc7',newOutputLayer);%% 训练配置options=trainingOptions('adam',...'MaxEpochs',15,...'MiniBatchSize',64,...'InitialLearnRate',0.001,...'ValidationData',imdsTest,...'ValidationFrequency',30,...'Plots','training-progress',...'ExecutionEnvironment','gpu');%% 训练模型net=trainNetwork(augimdsTrain,lgraph,options);%% 评估模型YPred=classify(net,imdsTest);accuracy=sum(YPred==imdsTest.Labels)/numel(imdsTest.Labels);fprintf('准确率:%.2f%%',accuracy*100);%% 可视化混淆矩阵figure;cm=confusionchart(imdsTest.Labels,YPred);cm.Title='混淆矩阵';cm.ColumnSummary='column-normalized';