【树叶识别】基于matlab卷积神经网络树叶识别系统【源码46期】

一、简介

基于matlab卷积神经网络树叶识别系统，可以识别7种不同类型的图像树叶。卷积神经网络（Convolutional Neural Networks, CNNs）已成为图像识别领域的核心方法。由于其出色的特征提取能力和对图像局部信息的有效处理，CNNs在树叶识别等任务中展现出了显著的优势。本文将介绍如何使用MATLAB和CNN实现树叶识别，并探讨其在实际应用中的潜力和挑战。

二、部分源码

clear;clc;
%% 加载数据
allImages = imageDatastore('dbx', ...
    'IncludeSubfolders' ,true, ...
    'LabelSource' , 'foldernames' );% 图像加载为图像数据存储
% imageDatastore函数会根据文件夹名称自动标记图像
% 划分训练集(80%)和测试集(20%)
[imgsTrain,imgsTest] = splitEachLabel(allImages,0.8,'randomized');
% 显示训练集和测试集的数量
disp(['Number of training images: ',num2str(numel(imgsTrain.Files))]);
disp(['Number of validation images: ',num2str(numel(imgsTest.Files))]);

%% 构建CNN
layers = [ ...
    imageInputLayer([227 227 3]) % 图像输入层，输入维度为256*256*3
    convolution2dLayer([3 3],32) % 卷积核大小7x7、数量32
    batchNormalizationLayer
    reluLayer
    maxPooling2dLayer([7 7],'Stride',2) % 池化核大小、步长
    maxPooling2dLayer([3 3],'Stride',2)
    convolution2dLayer([3 3],32)
    batchNormalizationLayer
    reluLayer
    maxPooling2dLayer([3 3],'Stride',2)
    fullyConnectedLayer(8)  % 全连接层，输出为2维
    softmaxLayer
    classificationLayer]; % 分类层

options = trainingOptions('sgdm', ...% 使用sgdm优化器
    'ExecutionEnvironment','auto', ...% 自动选择运行环境，无GPU自动选CPU，有GPU自动选GPU
    'MaxEpochs',25,...% 批量
    'MiniBatchSize',20, ... % 每批的数据量
    'GradientThreshold',1, ...% 梯度阈值
    'Verbose',false, ... % 不在命令窗口显示学习过程

%% 开始训练
CNNnet = trainNetwork(imgsTrain,layers,options);
save CNNnet CNNnet; 
%% 测试集
y_pred = classify(CNNnet,imgsTest); % 使用训练好的网络测试
accuracy = mean(y_pred == imgsTest.Labels);% 计算准确率
disp(['总体准确率: ',num2str(100*accuracy),'%']);

三、运行结果

四、matlab版本

MATLAB R2019a

五、学习与交流

文中不足之处请大家多多指教，学习与交流可以联系企鹅：3752243968
文中部分源码仅供参考，若需要全部代码可以私信