我正在尝试使用转移学习方法对一组图像进行分类。我遇到的所有教程都使用Alexnet来微调和转移学习。但是,我试图使用像CIFAR-10这样不那么复杂的模型。我经历了这个Matlab Tutorial。在他们开始转移学习的时候,他们使用了Matlab的样本数据。他们将地面实况数据加载到表格中,
%Load the ground truth data
data = load('stopSignsAndCars.mat', 'stopSignsAndCars');
stopSignsAndCars = data.stopSignsAndCars;
该表包含停车标志,车辆前部和后部的图像文件名和ROI标签。每个ROI标签是图像内感兴趣对象周围的边界框。然后他们只保留停车标志的标签。
如果我想以类似于Alexnet的方式使用我自己的图像进行此网络,我该怎么办?
net=alexnet;
layersTransfer = net.Layers(1:end-3);
readAllDir=uigetdir('','Select All directory to read scaled image files');
AllSample=imageDatastore(readAllDir, 'IncludeSubfolders', true, 'LabelSource','foldernames');
通过类似的方法,我的意思是制作' imageDatastore'对于所有标记的图像并使用它们来训练网络。