本文快速的用MATLAB对自己的图像数据集进行训练和分类效果非常不错,利用预训练的alexnet网络对新图像集进行迁移学习分类准确率很高。
首先要知道AlexNet的输入层大小是2272273的,所鉯在输入自己的数据之前需要改变输入矩阵大小,下面定义一个IMAGERESIZE函数承担此任务
接着就是卷积神经网络的结构定义了,通过imageDatastore函数将数據集读取
%第一个参数./images表示文件所在的路径; %includesubfolders:是否继续读取子文件夹中的图像数据;与后面的true相对应
其分类准确率如图所示: