如何使用Matlab通过预先培训的CIFAR-10网络传输学习?

时间:2017-08-08 21:58:42

标签: matlab machine-learning classification conv-neural-network

我正在尝试使用转移学习方法对一组图像进行分类。我遇到的所有教程都使用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'对于所有标记的图像并使用它们来训练网络。

0 个答案:

没有答案