matlab
花代码MatLab-深度学习
Matlab
上的高级机器学习示例
迁移学习示例脚本
下面的代码实现了本章花种示例的迁移学习。
它在课程示例文件中作为脚本
trainflowers.mlx
提供。
您可以从右上角的帮助菜单下载课程示例文件。
您可以在牛津大学的页面上找到有关此数据集的更多信息。
请注意，如果您在不支持
GPU
的计算机上运行此示例，则它可能需要一些时间才能运行。
获取训练图像
flower_ds
=
imageDatastore(
'
Flowers
'
,
'
IncludeSubfolders
'
,
true
,
'
LabelSource
'
,
'
foldernames
'
);
[trainImgs,testImgs]
=
splitEachLabel(flower_ds,
0.6
);
numClasses
=
numel(categories(flower_ds.Labels));
通过修改
AlexNet
创建网络
net
=
alexnet;
layers
=
net.Layers;
layers(
end
-
2
)
=
fullyC
